1、下載安裝python,勾選添加到環境,選擇自定義安裝,選擇路徑,其他默認。安裝完成后,pip也安裝完畢(包含在python中)。以管理員身份運行cmd,輸入:pip --version,查看pip版本,有版本號則說明安裝成功。
2、更換源,win+R 輸入%HOMEPATH%,在此目錄下創建 pip 文件夾,其內創建 pip.ini 文件,粘貼如下內容:
[global] timeout = 6000 index-url = https://pypi.tuna.tsinghua.edu.cn/simple trusted-host = pypi.tuna.tsinghua.edu.cn
3、PyTorch官網查看支持的cuda版本,下載支持的cuda和對應的cudnn。
安裝cuda,默認路徑,安裝完成后,把下載的cudnn解壓,放到cuda安裝路徑的對應文件夾中。
在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite目錄中,按下shift+鼠標右鍵,在此處打開powershell窗口,輸入./bandwidthTest.exe,出現GPU信息則安裝成功。
4、安裝pytorch(pip命令行方式),以管理員身份運行cmd,復制粘貼步驟3圖中的代碼,回車即可。若出現安裝失敗,可以逐個安裝,
先 pip install torch===1.4.0 -f https://download.pytorch.org/whl/torch_stable.html
再 pip install torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html
完成后,在cmd中輸入紅框標示的代碼,信息如圖則說明pytorch安裝成功,且可以使用GPU。