conda很容易裝,安裝百度到的教程一般都不會錯。
重複三遍,cuda的版本是通過nvcc -V看到的,cuda的版本是通過nvcc -V看到的,cuda的版本是通過nvcc -V看到的,不是 nVidia-smi
有個地方要注意,就是conda帶的python的版本要和你代碼一致,比如這次juji 的任務就是要python3.7的,博世指定。參考:https://blog.csdn.net/yuejisuo1948/article/details/81043823和https://blog.csdn.net/yuejisuo1948/article/details/81043823
但是pytorch,換國內原,國外默認源,都不行,都是 段錯誤 (核心已轉儲),甚至 pipinstall 國內各源都失敗,肯定在線安裝網速導致裝不上了,找到另外一個辦法,就是離線安裝pytorch,百度一下,果然
下面這篇推薦 說離線安裝。
https://blog.csdn.net/felicityPY/article/details/90233374
進入下面這個地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
根據nvcc -V查到的自己cuda 是10.0,就對應的下載 pytorch-1.4.0-py3.8_cuda10.0.130_cudnn7.6.3_0.t.
好像上面下載的包安裝出現說破損,
然後我又去 https://download.pytorch.org/whl/cu100/torch_stable.html 官網離線安裝包下載
還是下面這樣子速度快
上面那個下載地址是在
https://pytorch.org/get-started/previous-versions/
然後我們可以打開終端,使用conda install 命令即可,記得先激活你的安裝環境
完了之後用什麼的命令安裝
pytorch就安裝成功了。
最後我們進入python環境測試一下即可:
最終顯示成功