今天瘋狂踩坑后的記錄分享
報錯資訊
GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
解決方法
環境：Win10+Python3.6.12+CUDA11.1+Cudnn8.0.4
- CUDA11.1和Cudnn8.0.4從官網安裝
- 使用
conda install pytorch torchvision cudatoolkit=11 -c pytorch-nightly來安裝pytorch - 使用tensorflow的話 可以先
conda install cudatoolkit=11然后再pip install tensorflow-gpu==1.15
先裝好tf就無法安裝cudatoolkit11，conda自帶的是cudatoolkit10；
先裝好cudatoolkit11，再用conda裝tf可能會報錯，用pip不會
注意
由于需要cuda11的支持，因此向下兼容的原則不適用，
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/168165.html
標籤：其他
上一篇：做站外包給三方需要注意哪些問題