大家好，又見面了，我是你們的朋友全棧君。
安裝步驟
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
安裝成功
通過上面輸出的資訊，說明安裝是成功的
釋出者：全棧程式員棧長，轉載請注明出處：https://javaforall.cn/152754.html原文連結：https://javaforall.cn