配置環境總體思路
1.依據python版本選擇對應Anaconda版本;
2.依據顯卡驅動版本選擇對應的CUDA版本;
3.依據CUDA版本選擇對應的cudnn和pytorch版本。
一、Anaconda安裝
1.下載地址
1.官網 https://www.anaconda.com/products/individual
2.清華源 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
注:若官網下載速度慢,可用清華源下載;注意操作系統的位數(一般是64位的)
2.安裝教程
1.雙擊運行下載的.exe文件,按照以下順序點擊:Next → I Agree
2.選擇用戶:Just Me 或 All Users 都可以,默認選擇 Just Me
3.選擇安裝目錄:一般不建議安裝到C盤;可更改安裝路徑,注意安裝路徑不能出現中文、空格,使用純英文即可。
4.勾選安裝參數:一是添加anaconda到環境變量中(path),二是將anaconda的python版本作為默認的python版本; 建議兩者都勾選,環境變量手動再添加一遍(后續有介紹)。
5.等待安裝即可,結束時最后 “兩個learn” 都不需要勾選,點擊Finish即可。
3.配置環境變量
1.鼠標選擇”此電腦“,單擊鼠標右鍵,選擇 ”屬性“;
2.左鍵單擊左側欄的 “高級系統設置”;
3.點擊 “高級”欄中的 “環境變量”;
4.點擊 “系統變量”中的 “Path”項,選擇編輯;
5.點擊 “新建”,然后添加以下類似路徑(以安裝路徑為准,只需保證最后路徑的文件名稱相同即可)
E:\Anaconda (Python需要)
E:\Anaconda\Scripts (conda自帶腳本)
E:\Anaconda\Library\mingw-w64\bin (使用C with python的時候)
E:\Anaconda\Library\bin (jupyter notebook動態庫)
E:\Anaconda\Library\usr\bin (我自己沒找到此文件夾,若存在可將此條也加上)
注:切記點擊確定!!
4.安裝測試
1. win+R,輸入cmd,進入命令提示符
2. 輸入: conda --version
若跳出conda版本,則表示anaconda安裝成功,且已經配置好環境變量
二、CUDA安裝
1.確定版本
控制面板——硬件和聲音——NVIDIA控制面板——幫助——系統信息——組件
1.第三行,NVCUDA64.DLL 可查看顯卡驅動版本支持的最高CUDA版本;
2.網頁搜索相應CUDA版本安裝,建議不需要安裝太新版本。網頁直接可查找;
2.安裝步驟
1.雙擊運行.exe文件后,前兩項是默認安裝;
2.當選擇安裝選項時,選擇自定義安裝,驅動程序組件第一次安裝盡量全選;
3.記得留意CUDA安裝位置,之后cudnn的安裝會用到!!
3.測試
1.win+R,輸入cmd,進入命令提示符
2.輸入:nvcc -V
4.添加至系統變量
1. 添加方法與anaconda的相同:**此電腦——屬性——高級系統設置——環境變量——path——編輯——新建**
2.添加以下類似路徑:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp
3.有些路徑在安裝時可能已經添加,手動補充沒有的即可
三、cuDNN配置
1.進入官網依照cuda版本選擇cudnn文件下載(需要注冊賬號)
https://developer.nvidia.cn/rdp/cudnn-download
2.解壓下載的cudnn文件,會獲得三個文件夾:bin、include、lib;
3.將這三個文件夾復制到之前CUDA的安裝路徑里,最后一個路徑名應是cuda版本號
4.CUDA安裝時若是默認路徑,則一般為:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
四、pytorch安裝(清華鏡像)
1.打開 Anaconda Prompt
2.添加清華鏡像
依次輸入下列三行代碼:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
3.添加pytorch鏡像
接着輸入下列代碼:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
4.安裝pytorch
進入官網鏈接: https://pytorch.org/
1.在官網根據自己之前安裝的環境依次選擇,然后復制官網給出的命令語:
Install——stable——Windows——conda——python——10.2 (以cuda10.2為例)
2.若按照上列方式官網提供的cuda版本過高,點擊上方的 “ install previous versions of pytroch”,選擇舊版本
3.根據不同的版本搜索自己的下載語句,記住去掉 -c pytorch
conda insatll pytorch torchvision torchaudio cudatookit=10.2 (以cuda10.2為例)
5.測試
1.win+R,輸入cmd,進入命令提示符
2.輸入
python # 進入>>>python環境
import torch # 如正常則靜默
a = torch.Tensor([1.]) # 如正常則靜默
a.cuda() # 如正常則返回"tensor([ 1.], device='cuda:0')"
from torch.backends import cudnn # 如正常則靜默
cudnn.is_acceptable(a.cuda()) # 如正常則返回 "True"
6.注意事項
如果以前安裝過pytorch或者安裝失敗過,在安裝前執行下面兩條語句:
conda uninstall pytorch
conda uninstall libtorch