Windows安裝Pytorch并配置Anaconda與Pycharm
1 開發環境準備
Python 3.7+Anaconda3 5.3.1(64位)+CUDA+Pycharm Community
2 安裝Anaconda
2.1 進入官網下載:
根據windows版本選擇64位或32位進行下載,這里選擇3.7版本64位進行下載。

2.2 下載后獲得.exe文件,雙擊運行
2.2.1 點擊“Next”

2.2.2 點擊“I Agree”

2.2.3 點擊“Next”

2.2.4 默認安裝路徑,點擊“Next”

2.2.5 注意勾選“Add Anaconda3 to my PATH environment variable”,再點擊“Install”

2.2.6 等待安裝完成(時間約為3~5分鐘)

2.2.7 點擊“Next”

2.2.8 點擊“Next”

2.2.9 點擊“Finish”,安裝完成

2.3 測試是否安裝成功
按住鍵盤上的“win+R”進入windows運行,輸入“cmd”進入命令行窗口,輸入命令“conda list”。如果出現以下很多庫文件名稱,則表示Anaconda已經安裝成功。

3 安裝CUDA
3.1 查看支持CUDA的GPU型號
安裝CUDA的必要條件之一就是要求電腦擁有Nvidia顯卡,同時需要提前查看支持CUDA的GPU型號。
3.2 進入官網下載
這里選擇目標平臺的配置為“Windows+x84_64+10+exe[local]”,參照下圖進行勾選,點擊“Download”下載文件到本地。

3.3 下載后獲得.exe文件,雙擊運行
3.3.1 點擊“OK”
選擇默認路徑,點擊“OK”進行安裝。


3.3.2 點擊“Next”
等待約10分鐘,出現以下界面,點擊“同意并繼續”。

3.3.3 點擊“下一步”
選擇“精簡”,點擊“下一步”。

3.3.4 點擊“NEXT”
勾選“I understand,……”,點擊“NEXT”。安裝完成后,界面自動消失。

3.4 確定安裝成功
3.4.1 安裝確認
找到安裝的文件夾,這里默認是“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin”,查看是否存在“nvcc.exe”文件。

3.4.2 安裝測試
按住鍵盤上的“win+R”進入windows運行,輸入“cmd”進入命令行窗口,輸入命令“nvcc -V”。如果出現下圖版本信息,則表示CUDA已經安裝成功。

4 安裝Pytorch
4.1 進入官網
進入官網,選擇首選項為“Stable(1.4)+Windows+Conda+Python+10.1”,即可獲得安裝的命令“conda install pytorch torchvision cudatoolkit=10.1 -c pytorch”。

4.2 下載后獲得.exe文件,雙擊運行
4.2.1 以管理員身份打開命令提示符
點擊“開始”,找到“Windows系統”里的“命令提示符”,選擇“以管理員身份運行”即可打開對話框。

4.2.2 將4步驟4.1所得到的命令復制到命令提示符界面中(不建議使用這種方法)
由于國內網絡的限制,不建議直接使用此方法進行下載安裝,可直接跳過步驟4.2.2直接轉到步驟4.2.3進行操作。

等待電腦環境自動檢測完成

在彈出“Proceed([y]/n)?”后,輸入“y”并回車即可進行下載安裝。

4.2.3 更換國內源(如清華鏡像源)進行下載安裝
在已經打開的命令行中輸入以下代碼:
1 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
2
3 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
4
5 conda config --set show_channel_urls yes
6
7 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
8

然后將Pytorch官網上獲取的代碼復制到命令行中,并刪除后面的“-c pytorch”,回車即可進行環境的自動檢測。
1 conda install pytorch torchvision cudatoolkit=10.1

在彈出“Proceed([y]/n)?”后,輸入“y”并回車即可進行下載安裝。

過一段時間后,安裝自動完成,界面如下圖所示。

5 安裝Pycharm
5.1 進入官網
進入官網,選擇右側免費的社區版(Community)進行下載。

5.2 下載后獲得.exe文件,雙擊運行
5.2.1 點擊“Next”

5.2.2 默認安裝路徑,點擊“Next”

5.2.3 選擇安裝選項,點擊“Next”
勾選“64-bit launcher”、“.py”后,點擊“Next”。

5.2.4 點擊“Install”

5.2.5 等待安裝

5.2.6 點擊“Finish”,安裝完成

5.2 編譯環境配置
5.3.1 打開Pycharm,點擊“Create New Project”

5.3.2 默認工程名,點擊“Create”

5.3.3點擊“File”→“Create”→“Settings...”

5.3.4點擊“Project Interpreter”
勾選“Project Interpreter”,點擊設置按鈕,選擇“Show All…”。


5.3.5 點擊“Project Interpreter”
勾選“Existing environment”,選擇“C:\Users\Wang Qi\anaconda3\python.exe”,如未顯示則按此路徑尋找選中python.exe。勾選“Make available to all projects”,點擊“OK”、“OK”保存退出即可。

5.3.6 新建main.py文件
選中當前工程文件夾“untitled”,右擊選擇“New”→“Python File”,隨意命名一個文件即可,這里命名為“main”。


5.3.7 測試Pytorch是否安裝成功
在“main.py”文件中輸入以下代碼并運行:
1 import torch
2
3 print(torch.__version__)
4 print('gpu:', torch.cuda.is_available())
如果輸出如下圖所示的內容,則表示Pytorch安裝成功。

至此,Windows安裝Pytorch并配置Anaconda與Pycharm所有工作都已完成,讓我們一起開啟Pytorch的學習使用之旅吧!
參考資源: