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
至此,安裝成功!