win10中pyTorch的GPU模式安裝記錄
pyTorch的GPU模式安裝記錄
pyTorch \qquad\text{pyTorch} pyTorch的 CPU \text{CPU} CPU 模式安裝比較簡單,可以忽略第1,2節,直接採用第3節的方式。
pyTorch
\qquad\text{pyTorch}
pyTorch的
GPU
\text{GPU}
GPU 模式需要先安裝
CUDA
\text{CUDA}
CUDA 和
cuDNN
\text{cuDNN}
cuDNN,然後才安裝
pyTorch
\text{pyTorch}
pyTorch。
\qquad
1. 安裝CUDA
\quad
進入到 CUDA Toolkit Archive 選擇想要下載的
CUDA Toolkit
\text{CUDA Toolkit}
CUDA Toolkit版本:
\qquad
\quad
由於目前
pyTorch
\text{pyTorch}
pyTorch 的
whl
\text{whl}
whl 檔案只支援到11.0版本(見第 3 節),因此選擇cuda_11.0.2_451.48_win10
:
\qquad
\qquad
2. 安裝cuDNN
\quad
進入到 cuDNN Download 選擇想要對應的
cuDNN
\text{cuDNN}
cuDNN 版本(需要註冊nvidia會員登陸才能下載):
\qquad
\quad
由於
CUDA Toolkit
\text{CUDA Toolkit}
CUDA Toolkit 選擇了11.02版,因此
cuDNN
\text{cuDNN}
cuDNN 選擇對應的8.0.5版cudnn-11.0-windows-x64-v8.0.4.30
:
\qquad
\quad
將cudnn-11.0-windows-x64-v8.0.4.30解壓後的cuda資料夾下的3個目錄(bin,include,lib)拷貝到中C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0中:
\qquad
\quad
然後,就可以使用nvcc -V
檢視
CUDA
\text{CUDA}
CUDA是否安裝成功:
\qquad
\qquad
3. 安裝pyTorch
\quad
進入pyTorch官網
\qquad
(
1
)
(1)
(1) 採用
pip
\text{pip}
pip 安裝方式
\qquad
pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
\quad
(
2
)
(2)
(2) 安裝過程中需要下載torch-1.7.1+cu110-cp37-cp37m-win_amd64.whl
檔案,由於該檔案很大(1.9 G),可以直接到https://download.pytorch.org/whl/torch_stable.html下載:
\qquad
\quad
採用
pip
\text{pip}
pip 安裝方式:
\qquad
pip install torch-1.7.1+cu110-cp37-cp37m-win_amd64.whl
\qquad
pip install torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
經過這3個步驟之後,還是經常會出現CUDA安裝不成功,torch.cuda.is_available() 測試結果為 False
主要問題在於顯示卡驅動的版本問題。
\quad
4. 顯示卡驅動設定
\quad
如果是雙顯示卡,必須到“NVIDIA控制皮膚”中選擇“高效能NVIDIA處理器”。
\qquad
\quad
在“NVIDIA控制皮膚”的“幫助”—“系統資訊”中可以看到“驅動版本”和該版本所支援的CUDA版本號:
\qquad
\qquad
安裝最新版的驅動程式(顯示卡GTX850M,460.89-notebook-win10-64bit-international-dch-whql),顯示卡驅動支援CUDA Toolkit12.0,因此特意找了一個稍微老一點的驅動(451.67-notebook-win10-64bit-international-dch-whql)才能支援CUDA Toolkit11.02,否則無法下載到可以匹配的CUDA Toolkit版本。
顯示卡驅動下載地址:https://www.nvidia.cn/Download/index.aspx?lang=cn
\quad
\quad
在NVIDIA CUDA Toolkit Release Notes中,可以查到CUDA Toolkit和顯示卡驅動的匹配關係:
\qquad
測試CUDA是否安裝成功
\quad
安裝完畢之後,在
python
\text{python}
python 中可以檢視已安裝
pyTorch
\text{pyTorch}
pyTorch 和
CUDA
\text{CUDA}
CUDA 的版本:
\qquad
\quad
在
python
\text{python}
python 檢視
CUDA
\text{CUDA}
CUDA 是否安裝成功:
\qquad
後記
\quad 本文只是記錄了一個大概可行的流程,在Win10中的三次安裝記錄如下:
第一次安裝:首先更新顯示卡驅動到最新的460.89,然後安裝pyTorch+cu10.2,最後安裝CUDA和cuDNN10.2的版本
但是CUDA安裝沒成功: torch.cuda.is_available() 結果為 False
第二次安裝:把顯示卡驅動降到了451.67以匹配CUDA 11.02。然後重新安裝了pyTorch+cu11.02,雖然之前安裝的CUDA Toolkits和cuDNN是10.2的版本
但是CUDA安裝卻測試成功: torch.cuda.is_available() 結果為 True
第三次測試:刪掉了CUDA和cuDNN10.2的版本,安裝了CUDA和cuDNN11.02的版本,CUDA安裝測試成功: torch.cuda.is_available() 結果為 True
\quad
由於測試較少,準確的安裝過程不是特別肯定,似乎是:
(
1
)
(1)
(1) 應該先安裝CUDA和cuDNN,然後再安裝pyTorch;
(
2
)
(2)
(2) CUDA Toolkit和顯示卡驅動版本的匹配很重要(不明白為何最新的顯示卡驅動所支援的CUDA Toolkits版本那麼高,只能通過降低顯示卡驅動版本,來匹配NVIDIA官網的CUDA Toolkit下載)。
相關文章
- Pytorch的GPU版本安裝PyTorchGPU
- 安裝pytorch-gpu的經驗與教訓PyTorchGPU
- Windows10安裝Pytorch步驟記錄WindowsPyTorch
- 解決pytorch-gpu 安裝失敗方案PyTorchGPU
- Pytorch中backward()的思考記錄PyTorch
- Torch GPU版本的安裝GPU
- 記錄Win10安裝MySQL5.7的步驟Win10MySql
- 安裝pytorchPyTorch
- pytorch安裝PyTorch
- Docker安裝記錄Docker
- lepus安裝記錄
- HandlerSocket安裝記錄
- Mantis安裝記錄
- win10 安裝CUDA、cudnn、tensorflow-gpuWin10DNNGPU
- windows下pytorch的安裝WindowsPyTorch
- pytorch安裝命令PyTorch
- pytorch安裝教程PyTorch
- Win10 平臺下, LightGBM GPU 版本的安裝Win10GPU
- 記錄 Ubuntu 安裝 homestaedUbuntu
- Redis On Windows安裝記錄RedisWindows
- Macbook安裝Cplex記錄Mac
- 安裝seafile記錄文件
- Centos 安裝Mysql記錄CentOSMySql
- Ubuntu安裝使用記錄Ubuntu
- tensorflow-gpu 安裝GPU
- mac os x中安裝mysql+phpmyadmin的方法記錄MacMySqlPHP
- rosbridge簡單的安裝記錄ROS
- Pytorch環境安裝PyTorch
- Pytorch安裝問題PyTorch
- 安裝PyTorch_1PyTorch
- Pycharm 個人安裝小記錄PyCharm
- Jenkins 安裝簡記錄Jenkins
- 安裝apache+php記錄ApachePHP
- nagios snmp 安裝記錄iOS
- Linux工具安裝記錄Linux
- c++ protobuf安裝記錄C++
- PyTorch深度學習入門筆記(一)PyTorch環境配置及安裝PyTorch深度學習筆記
- docker安裝tensorflow-gpuDockerGPU