以前我是直接在colab中使用pytorch,非常方便,今天折騰了一上午,終于搞定了pytorh的安裝和環境設置,分享下我的安裝流程,遇到的問題和解決方案。
1. 用pip安裝Pytorch
打開cmd窗口(按win + R),執行下列語句
pip install torch torchvision
執行時先后出現了兩個錯誤:
(1). “ValueError: check_hostname requires server_hostname”
解決方法:關掉VPN
(2). “ERROR: Could not install packages due to an OSError: [WinError 5] 拒絕訪問。: 'c:\\python39\\Library'
Consider using the `--user` option or check the permissions.”
問題原因:PIP安裝模塊的權限不夠導致失敗
解決方法:加--user??
pip install --user torch torchvision
2. 配置環境變量
在安裝完PyTorch后,需要將PyTorch的路徑添加到系統環境變量中:
在開始菜單中,搜索“系統”:
在系統信息中找到“高級系統設置”,點擊進入:
點擊“環境變量”,再在彈出來的面板中找到“系統變量”>Path,點擊“新建”
將PyTorch安裝目錄下的“lib”文件夾的路徑添加進去
注意具體的路徑要根據自己的實際安裝路徑進行修改,可以在開始菜單中搜索“torch”來找到相關文件夾,尋找相同后綴的地址。
3. 檢驗是否安裝成功,查看版本
import python
出現錯誤:“ 'import' is not recognized as an internal or external command, operable program or batch file.”
解決方法:先輸入python語句,再import torch
然后可以查看torch版本
print(torch.__version__)
(注意“__”是兩個短下劃線)
4. 在vscode中使用pytorch
注意一定要在vscode中選擇安裝了torch的環境:
輸入:
import torch
但又出現以下問題:
ImportError:failed to load Pytorch C extensions:It appears that Pytorch has loaded the ‘torch/_C’ folder of the Pytorch repository rather than the C extensions which are expected in the ‘torch._C’ namespace.
原因:numpy庫重復。環境里本來有一個numpy庫,但裝torch的時候又附加了另一個版本的numpy庫,調用的時候出現調用錯亂的問題。
解決方法:在anaconda prompt中激活所用環境,用pip卸載一個numpy。
成功啦!
下面,就可以真正地探索pytorch啦,比如,創建個張量:
x = torch.empty(3, 2)
print(type(x))
print(x)
感覺安裝是學習一個庫/軟件等內容中最難推進的一項了,有時似乎真的是玄學,要不停地試很多種方法,解決了一個問題,又出現了各種各樣奇奇怪怪的error,但要相信只要不放棄,總能解決的,祝大家成功!