安裝步驟

1.檢查是否有合適的GPU, 若有安裝Cuda與CuDNN

2.下載pytorch安裝文件，進行pytorch的安裝

檢查電腦是否有合適的GPU

在桌面上右擊如果能找到NVIDA控制面板，則說明該電腦有GPU?？刂泼姘迦缦?#xff0c;并通過查看系統信息獲取支持的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

?

安裝成功

通過上面輸出的信息，說明安裝是成功的

總結

以上是生活随笔為你收集整理的Pytorch(一) Pytorch 的安装的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。