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的學習使用之旅吧!
參考資源: