cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
生活随笔
收集整理的這篇文章主要介紹了
cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
在網上查了很多資料,首先要確定是不是版本問題,
像我在虛擬環境pip安裝用pip list查看的cuda和torch分別是10.2.89和1.5.0
但是在cuda list里面是10.2.89和1.0.0
很怪,無法理解
但是在python中打印出cuda和torch又分別是9.0和1.0.0
5555好窒息,搞了一整天很暴躁
conda list
在python中打印
無法理解
conda list和pip list的版本不同又和python打印出來的不同,而且我配置的torchvision是0.4的,肯定不能支持cuda9.0
具體可參考 https://www.zhihu.com/question/395145313/answer/1260811851
好吧,強行重裝cuda和torch
conda install pytorch=1.5.0 cudatoolkit=10.2.89 -c pytorch再次在python里面打印出來,
import torch print(torch.__version__) print(torch.cuda.is_available()) torch.version.cuda
成功!再次運行,報錯解除!
總結
以上是生活随笔為你收集整理的cuDNN error: CUDNN_STATUS_EXECUTION_FAILED的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: permission denied和li
- 下一篇: 加载torchvision中预训练好的模