电脑环境是win7,安装完CUDA和CUDNN之后,运行会出现上述错误。

RuntimeError: cuDNN version mismatch: PyTorch was compiled against 7003 but linked against 6021。

发现网上只有针对linux的改动方法,说是去除环境变量,而自己也消除了环境变量还是不好使。于是自己一步一步摸索,发现是cudnn64_7.dll和cudnn64_6.dll两个文件同时存在时可能会出现此问题。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\bin

D:\anaconda\Lib\site-packages\torch\lib

一般会存在与这两个地址下,只要删除cudnn64_6.dll即可保证正常运行。

当然也可以在电脑上搜索cudnn64_6,把搜索到的删除也可以。



Logo

更多推荐