1.安裝conda(Miniconda、Anaconda都可以)
以下默認的是python3.8,pytorch選擇的是CPU版本!實際可以根據需求安裝任意版本。如果CUDA不匹配的可以卸載原來的CUDA安裝新的CUDA。
2.GPU版本需要確定CUDA版本
在桌面點擊右鍵,選擇“NVIDIA控制面板”
依次點擊:幫助-->系統信息
點擊“組件”,如圖所示,本電腦CUDA是11.4的
3.下載Pytorch的離線包
推薦網站:https://download.pytorch.org/whl/torch_stable.html
技巧:可以打開pytorch的官網PyTorch查看自己適合什么版本(注意選擇CUDA版本!),然后再進入上面的網站單獨下載
如:
對應的分別下載(%2B應該是beta版本,實際可以下載正式版本):
最后下載得到的有:
3.安裝
- 首先打開cmd
- 使用“conda activate xxxxx(之前創建的環境名字)”激活到當前環境
- 再使用cd定位到下載torch的目錄中
- 最后依次pip install torch、torchvison、torchaudio(一定要輸入完整的名字)
- 全部安裝完了試一試輸入python進入解釋器,再輸入import torch。如果沒有報錯即安裝成功。
- 附:安裝過程中可能有些包的下載速度有些慢,可以使用清華源替換.condarc這個文件,詳細參考清華源的界面介紹