Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Invalid argument: Default MaxPoolingOp only supports NHWC on device type CPU #12

Open
Mr-doraemon opened this issue Jul 6, 2020 · 10 comments

Comments

@Mr-doraemon
Copy link

Mr-doraemon commented Jul 6, 2020

Hi, when I try to run python3 tracking/do_tracking.py --main main_otb2015, the following error will occurs.

**`E tensorflow/core/common_runtime/executor.cc:642] Executor failed to create kernel. Invalid argument: Default MaxPoolingOp only supports NHWC on device type CPU
[[{{node pool0/MaxPool}}]]

Traceback (most recent call last):
File "/home/users2/anaconda3/envs/env3.6/lib/python3.6/site-packages/tensorflow_core/python/client/session.py", line 1365, in _do_call
return fn(*args)
File "/home/users2/anaconda3/envs/env3.6/lib/python3.6/site-packages/tensorflow_core/python/client/session.py", line 1350, in _run_fn
target_list, run_metadata)
File "/home/user2/anaconda3/envs/env3.6/lib/python3.6/site-packages/tensorflow_core/python/client/session.py", line 1443, in _call_tf_sessionrun
run_metadata)
tensorflow.python.framework.errors_impl.InvalidArgumentError: Default MaxPoolingOp only supports NHWC on device type CPU
[[{{node pool0/MaxPool}}]]
**

I know some people say that because TensorFlow uses CPU version, but I only installed the GPU version of TensorFlow and the version is 1.15. Could you help me solve this problem?

@pvoigtlaender
Copy link
Contributor

Hi,

TensorFlow is not using the GPU for some reason. Please first make sure that you are able to use the GPU at all with the current TensorFlow version, there should be some explanation on how to find out which device is used on the TensorFlow website/Google.

@EchoAmor
Copy link

EchoAmor commented Jul 9, 2020

Hi,I met this problem too,have u solved it? can u share it with me please? Thanks very much @Mr-doraemon

@Mr-doraemon
Copy link
Author

Hi,I met this problem too,have u solved it? can u share it with me please? Thanks very much @Mr-doraemon

I think you should make sure that your TensorFlow-GPU version fits your CUDA version

@EchoAmor
Copy link

EchoAmor commented Jul 9, 2020

@Mr-doraemon yes,I have googled this error,and found that it's because the CUDA's library can't be found, but I saw in tensorflow and make sure Tensorflow-GPU-1.15 fits with CUDA10.0+cudnn7.4, it's the same with mine. Still appearance the error... DO u use the same version?

@Mr-doraemon
Copy link
Author

Mr-doraemon commented Jul 9, 2020 via email

@EchoAmor
Copy link

EchoAmor commented Jul 9, 2020

Thanks anyway! If it works for you ,can u please tell me here ,thanks very much! @Mr-doraemon

@Mr-doraemon
Copy link
Author

Mr-doraemon commented Jul 9, 2020 via email

@EchoAmor
Copy link

EchoAmor commented Jul 9, 2020

I have solved this by add environment variable
debug_cuda
but get another error--

`tensorflow.python.framework.errors_impl.InternalError: 2 root error(s) found.

(0) Internal: cuDNN launch failure : input shape([1,3,805,1093]) filter shape([7,7,3,64])
[[node conv0/Conv2D (defined at /home/echo/anaconda3/envs/Siamrcnn/lib/python3.6/site-packages/tensorflow_core/python/framework/ops.py:1748) ]]
[[rpn/feature/_2535]]
(1) Internal: cuDNN launch failure : input shape([1,3,805,1093]) filter shape([7,7,3,64])
[[node conv0/Conv2D (defined at /home/echo/anaconda3/envs/Siamrcnn/lib/python3.6/site-packages/tensorflow_core/python/framework/ops.py:1748) `

I haven't change your code ,so I think it's maybe the version problem, you just give the Tensorflow version ,can you tell me what's the version of CUDA and CUDNN? @pvoigtlaender @Mr-doraemon

@Mr-doraemon
Copy link
Author

I have solved this by add environment variable
debug_cuda
but get another error--

`tensorflow.python.framework.errors_impl.InternalError: 2 root error(s) found.

(0) Internal: cuDNN launch failure : input shape([1,3,805,1093]) filter shape([7,7,3,64])
[[node conv0/Conv2D (defined at /home/echo/anaconda3/envs/Siamrcnn/lib/python3.6/site-packages/tensorflow_core/python/framework/ops.py:1748) ]]
[[rpn/feature/_2535]]
(1) Internal: cuDNN launch failure : input shape([1,3,805,1093]) filter shape([7,7,3,64])
[[node conv0/Conv2D (defined at /home/echo/anaconda3/envs/Siamrcnn/lib/python3.6/site-packages/tensorflow_core/python/framework/ops.py:1748) `

I haven't change your code ,so I think it's maybe the version problem, you just give the Tensorflow version ,can you tell me what's the version of CUDA and CUDNN? @pvoigtlaender @Mr-doraemon

now, I have the same problem as you!

@ValentiaSulli
Copy link

I have solved this by add environment variable
debug_cuda
but get another error--

`tensorflow.python.framework.errors_impl.InternalError: 2 root error(s) found.

(0) Internal: cuDNN launch failure : input shape([1,3,805,1093]) filter shape([7,7,3,64])
[[node conv0/Conv2D (defined at /home/echo/anaconda3/envs/Siamrcnn/lib/python3.6/site-packages/tensorflow_core/python/framework/ops.py:1748) ]]
[[rpn/feature/_2535]]
(1) Internal: cuDNN launch failure : input shape([1,3,805,1093]) filter shape([7,7,3,64])
[[node conv0/Conv2D (defined at /home/echo/anaconda3/envs/Siamrcnn/lib/python3.6/site-packages/tensorflow_core/python/framework/ops.py:1748) `

I haven't change your code ,so I think it's maybe the version problem, you just give the Tensorflow version ,can you tell me what's the version of CUDA and CUDNN? @pvoigtlaender @Mr-doraemon

This can be solved by conda install tensorflow-gpu=1.15.0

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants