RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
faller45 人气:0看我结论,大家试试看最后装pytorch看行不行,不行就去冲了PyTorch /Doge
ubuntu 20.04 下安装CUDA,参考这个博主写的,先看显卡支持的最高CUDA版本,之后找一个较新的不管是安装CUDA cuDnn 还是PyTorch都按一个版本装就行
不过最后的cudnn 我是在官网下的Deb安装的,然后参考这个链接里的 CUDA版本切换 这一步创建的那个cuda文件夹。
然后对MINIST数据集,跑一个简单的 卷积神经网络 的时候出现了RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED,找了一圈都没能解决
最后决定看看全连接能不能跑,报错:
RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling `cublasCreate(handle)
看到这就有希望了,最起码环境配置应该没问题,搜索后,也是被前面那个错误折腾的够呛,所幸降级PyTorch到1.7试试,然后就好了,,,(我pytorch刚开始是官网配置的,直接下)
难不成是因为我先安装的pytorch???不应该呀,反正之前的环境配的没问题,降级之后就好了,恩,这锅甩给pytorch 0.0
加载全部内容