工科女孩窮遊筆記

Windows 10 搭載 PyTorch 及 Anaconda

本來的顯卡是GTR 1050 memory 3GB 在使用上沒有問題，

但後來升級成 RTX 2080 memory 8GB

進行訓練時跳出以下兩種error

RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

Runtimeerror cublas runtime error the gpu program failed to execute at …

上網查詢後發現RTX系列的顯卡必須使用CUDA10，GTR系列則是都可以

所以必須把先前安裝給GTR的環境改掉..是一個不小的工程

藉此記錄下來以備不時之需

如果是從開始安裝可以跳過「卸載CUDA9」的步驟

原始環境（已安裝Anaconda）

Windows10 GPU: GTR1050

PyTorch 版本：1.1.0

Python版本：3.6.8

CUDA版本：9.0.176

cuDNN版本：7.6.1

升級後環境

Windows10 GPU: RTX2080

PyTorch 版本：1.2.0

Python版本：3.6.6

CUDA版本：10.0.13

cuDNN版本：7.6.2

若未安裝Anaconda，可至官網下載安裝檔案：https://www.anaconda.com/

下載後直接安裝就行了

• 環境版本查詢方法

首先進入Anaconda Prompt

CUDA:

nvcc -V

python: (進入python環境)

python

*退出python直接 Ctrl + Z 或是輸入

exit()

PyTorch:

import torch

print(torch.__version__)

CuDNN的查詢方法比較麻煩

進入你CUDA安裝的路徑

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include

用記事本開啟「cuDNN」的檔案

接著在裡面搜尋「CUDNN_MAJOR」

由此可知cuDNN的版本是7.6.1

－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－

因為RTX系列的顯卡要使用CUDA10，所以必須先把先前安裝的CUDA9卸載

• 卸載CUDA9

進入 控制台的程式集

除了「NVIDIA PhysX 系統軟體」和「NVIDIA圖形驅動軟體」其他都可以解除安裝

（不過我不知道為什麼我只有「NVIDIA PhysX 系統軟體」）

然後C槽內的C:\Program Files\NVIDIA GPU Computing Toolkit整個資料夾也可以刪除

都卸載乾淨之後就可以安裝CUDA10了

• 安裝CUDA10

首先進入NVDIA官網下載CUDA10

https://developer.nvidia.com/CUDA-10.0-download-archive?

Installer Type 要選local

下載之後就直接一直按下一步安裝

過程有遇到安裝失敗

原因是 Visual Studio Integration Failed

在自訂安裝裡把 CUDA 內的 Visual Studio Integration 勾選掉不要安裝

再次安裝就成功了

• 安裝cuDNN

進入NVIDIA官網下載（需要註冊）

https://developer.nvidia.com/rdp/cuDNN-download

選擇和你CUDA版本相同的cuDNN

我是CUDA10.0 所以選圖中紅色框框的

下載後解除安裝會有三個檔案

把這三個檔案複製後貼到

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0的資料夾裡面

就完成摟!

CUDA和cuDNN安裝完成後就可以開始安裝PyTorch了

安裝PyTorch

1. 建立虛擬環境（建議但非必要）

打開Anaconda Prompt後輸入

conda create --name pytorch python=3.6

*pytorch是環境名稱，可以自己任意取，python版本也可以改，

我是用3.6.6因為看到別人測試這個版本可以成功使用

*啟動虛擬環境 activate pytorch

*退出虛擬環境 deactivate

可以看到(base)變成(pytorch)代表已經在此環境裡

2. PyTorch安裝

若需要先卸載舊版本，則執行：

pip uninstall torch

（圖為安裝在基本環境base內不是pytorch，不過我兩個環境都有安裝）

安裝新版本PyTorch ↓↓↓

PyTorch 官網：https://pytorch.org/

進入官網的Get Started選取相對應版本的PyTorch

然後直接在Anaconda Prompt裡執行最下面的「Run this Command:」

conda install pytorch torchvision CUDAtoolkit=10.0 -c pytorch

（若想要安裝「舊版本的」PyTorch
可以來這裡尋找你想要安裝的版本： https://pytorch.org/get-started/previous-versions/ ）

這樣就安裝完成摟

要確認有沒有安裝完成就重複上述版本查詢動作

確認是否所有版本都正確

應該就大功告成可以執行了!

後記：

安裝兩次PyTorch的時間才間隔1個月PyTorch就出了新版本..

科技真的進步得好快