安裝步驟
1.檢查是否有合適的GPU, 若有安裝Cuda與CuDNN
2.下載pytorch安裝文件,進行pytorch的安裝
檢查電腦是否有合適的GPU
在桌面上右擊如果能找到NVIDA控制面板,則說明該電腦有GPU。控制面板如下,並通過查看系統信息獲取支持的Cuda版本
點擊 幫助->點擊 系統信息 彈出下面的對話框,在驅動程序版本那一欄就能看到該計算機使用的驅動版本。
下載Cuda
官網:https://developer.nvidia.com/cuda-10.1-download-archive-update2
在https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 這裏可以查詢到我們應該下載哪個版本
然後,根據實際情況選擇合適的版本
下載CuDNN
官網 https://developer.nvidia.com/rdp/cudnn-download
安裝Cuda
1.與安裝其他的軟件類似
2.安裝結束後將 ~/nvcc/bin(因爲版本的不同可能在不同的地方) 目錄添加到環境變量
3.在命令行下輸入 nvcc -V, 出現下列信息說明Cuda安裝成功
4.將CuDNN壓縮包解壓後,下面的三個文件夾複製到Cuda的安裝目錄下
5.然後執行Demo, 如果Demo中顯示PASS則說明安裝成功
安裝Pytorch
官網 https://download.pytorch.org/whl/torch_stable.html 選擇合適的版本
torch/torchvision 都需要安裝
驗證Pytorch
安裝成功
通過上面輸出的信息,說明安裝是成功的