PyTorch教程【一】Pytorch環境的配置及安裝


1、安裝Anaconda

詳細安裝教程參考我之前的博客:Anaconda的安裝

檢驗是否安裝成功:
(1)點擊開始菜單欄,打開Anaconda文件夾下的Anaconda Prompt

(2)在彈出的命令行窗口中看到base,即說明安裝成功

2、查看GPU型號及其驅動是否正常安裝

參考博客:查看GPU型號及其驅動是否正常安裝

3、有序的管理環境(使用Anaconda集成的conda包)

(1)開始菜單->Anaconda文件夾->Anaconda Prompt

輸入conda create -n pytorch python=3.6

  注:1、-n中的n是name
      2、pytorch可以替換成任意你想要起的名字(環境名)
      3、python=3.6 是該環境需要用到的包

輸入y

創建成功,提示激活方式

激活

激活成功,紅框處環境名稱顯示有變化

查看環境中的工具包:pip list

4、PyTorch的安裝

(1)進入PyTorch官網首頁:https://pytorch.org/

(2)下拉到安裝頁面

  PyTorch Bulid:一般選擇穩定版(Stable)
  Your OS:根據自己的操作系統進行選擇
  Package:主要涉及安裝方式,Windows系統一般選擇Conda,Linux系統一般選擇Pip
  Language:選擇Python
  CUDA:情況一:無英偉達顯卡或只有集顯,選擇None
        情況二:有英偉達顯卡且支持CUDA,推薦使用9.2
  Run this Command:安裝指令,復制后用來安裝

(3)檢查是否可以安裝CUDA

①、 查詢GPU是否支持CUDA:查詢GPU是否支持CUDA
②、查看GPU驅動版本,因為CUDA9.2以上只支持驅動版本大於392.26的,查看GPU驅動版本的方法見博客:Windows使用nvidia-smi查看GPU信息,驅動版本低於392.26需要進行升級,升級方法見博客:NVDIA的GPU驅動升級

附:CUDA的介紹見:CUDA

(4)打開Anaconda Prompt,切換到pytorch環境(新環境)

(5)切換conda源為清華鏡像源,參考博客:conda更換為清華鏡像源

(6)安裝cudnn(可能可以省略)

`conda install cudnn`

(7)修改Anaconda文件夾權限(不然后面可能會出問題)

找到Anaconda文件夾,右擊點擊屬性,點擊安全,選擇SYSTEM,點擊確定

(8)輸入復制的安裝命令,但是要把命令后綴的-c pytorch 刪除!!!

輸入y

(9)驗證是否安裝成功

輸入pip list,發現torch的版本

輸入python,輸入import torch,沒有報錯

輸入torch.cuda.is_available(),返回True

至此,安裝成功!


免責聲明!

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



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