Pytorch(一) Pytorch 的安裝

安裝步驟

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

 

安裝成功

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

發佈了41 篇原創文章 · 獲贊 5 · 訪問量 1萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章