看我結論,大家試試看最后裝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