pip安裝PyTorch(Windows系統)、cuda安裝


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。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM