Shakerato

pytorch CUDNN_STATUS_EXECUTION_FAILED error (Anaconda3) 본문

Research

pytorch CUDNN_STATUS_EXECUTION_FAILED error (Anaconda3)

Shakeratto 2019. 2. 23. 20:37
[previous setup]
conda install pytorch cuda90 -c pytorch 
(installed)
pytorch                   0.4.0           py36_cuda90_cudnn7he774522_1  [cuda90]  pytorch

-->

If you are using the latest GPU in the above environment, for example, 2080ti,
you may get an error message when you run pytorch code using GPU as shown in below.

CUDNN_STATUS_EXECUTION_FAILED

-->

[solution setup]
1. install cuda 10.0 and new cudnn for this
2. conda update -n base -c defaults conda
3. conda install pytorch cuda100 -c pytorch 
(installed)
pytorch                   1.0.1           py3.6_cuda100_cudnn7_1    pytorch


Comments