首先通過 python 官網將python安裝,這一步先跳過了
直接進入主題 torch 安裝
首先要進入 pytorch 官網(以給出)進入如下界面
點擊 install 進入到配置選擇的界面
Pytorch Build 這一行可供選擇的是 穩定版(左側)和 試用版 (右側)
your OS 這里選擇屬於自己的電腦系統
Package 這里需要選擇的是通過怎樣的途徑下載,本文以 pip 為例子進行安裝操作
CUDA 在自己電腦上的系統信息中查看電腦的CUDA型號,這里實際上是確認電腦是否支持 GPU 計算
在選擇好這些之后,在 Run this command 中會給出下載鏈接,將
pip install torch===1.7.0 torchvision===0.8.1 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
復制之后,win+R 輸入 cmd 打開,將復制內容粘貼到這里就可以開始下載安裝了
一般情況到這里就基本結束了,可以在命令行中進去python環境進行檢測,這樣說明安裝成功,就可以利用 pytorch 來構建自己的深度學習模型了
但是 往往安裝起來並不是那么簡單
最常見的錯誤就是“找不到模塊”“找不到安裝包”等錯誤,而一般情況下可以通過以下方法解決,因為存在 python 和 pytorch 模塊兼容問題,所以會存在python版本為3.7只能安裝torch1.5 和torchvision0.6,
所以如果你安裝的python版本是3.7或者之前的版本,那么在下載安裝 pytorch 時就要考慮下載合適的版本
法1:可以在 pytorch 官網復制那個下載鏈接是將
pip install torch===1.7.0 torchvision===0.8.1 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
改為
pip install torch===1.5.0 torchvision===0.6.1 torchaudio===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html
基本上就可以解決問題了
法2:通過如下鏈接下載對應的torch安裝包來直接下載安裝
https://download.pytorch.org/whl/torch_stable.html
更新1
問題:在調用 torch.from_numpy 時會出現 Module 'torch' has no 'from_numpy' memberpylint(no-member)的錯誤
解決:通過快捷鍵 Ctrl+Shift+P
打開vscode的命令面板,輸入settings.json
並打開第一項,將如下內容復制進去並保存,即可解決問題
"python.linting.pylintArgs": [
"--errors-only",
"--generated-members=numpy.*, torch.*, cv2.*, cv.*"
]
更新2
問題:vs code 運行 torch.rand() 時出現 “Module 'torch' has no 'rand' memberpylint(no-member)” 的錯誤
解決:通過快捷鍵 Ctrl+Shift+P
打開vscode的命令面板,輸入settings.json
並打開第一項,將如下內容復制進去並保存,即可解決問題
// whitelist torch to remove lint errors "python.linting.pylintArgs": [ "--extension-pkg-whitelist=torch", "--ignored-modules=torch", "--ignored-classes=torch", ]
pip 清華園安裝
-i https://pypi.tuna.tsinghua.edu.cn/simple