I can run demo.py with python demo.py on cpu ,but when i run on gpu, there is a bug :RuntimeError: CUDNN_STATUS_EXECUTION_FAILED , maybe the version of python,pytorch,cuda is incompatible,
this is my version what i use:
pytorch:0.3.0-cuda9.0
gpu:rtx-2080
cuda:9.0
python:3.6.7,
anyone know what's the problem? thanks
I can run demo.py with python demo.py on cpu ,but when i run on gpu, there is a bug :RuntimeError: CUDNN_STATUS_EXECUTION_FAILED , maybe the version of python,pytorch,cuda is incompatible, this is my version what i use: pytorch:0.3.0-cuda9.0 gpu:rtx-2080 cuda:9.0 python:3.6.7, anyone know what's the problem? thanks