一、一點前期准備
- 下載Anaconda,版本不限,下載完成后,點擊安裝時注意有“添加到環境變量”選項時記得勾選,這樣就可以不用自己再配置環境變量了。(如果你沒點,那么切記你的安裝路徑,后續自己添加環境變量時用的到,后續自己怎么添加,自行百度吧,如果點了,可以不用這一步的)
- 查看自己顯卡的版本號,方法為該鏈接; 找到自己版本號,如我的如下圖1;得到版本號后,找對應自己可以下載的cuda版本。如下圖2,你的版本號的值決定了你可以下載的cuda版本的上限,如我的可以選擇
CUDA 10.0.130版本及以下的cuda.


- 去該網址1和該網址2分別下載CUDA和CUDNN對應自己電腦的版本; 我選擇的是 cuda_9.2.148_win10和patch包以及cudnn7.2(cudnn版本號與cuda號是對應的),記住自己的cuda版本號!( 忘記了的話,可以用打開
cmd,輸入nvcc -V查看) - 點擊安裝CUDA及patch包;
- 然后將解壓Cudnn后得到的三個小文件夾(include,bin,lib)復制到CUDA的安裝目錄(默認安裝目錄C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2)即可;
- 在開始菜單里找到並打開 Anaconda promt,輸入conda --version以驗證conda是否安裝成功;
二、開始安裝pytorch
-
離線快速安裝方法,推薦 與網上直接聯網下載的方法不同,直接去清華鏡像網站下載到自己電腦,安裝會更加快速和安全。點擊該網址查看,找到自己對應版本的torch直接下載,很快,我的對應版本如下。

注意,torch的版本1.4.0不受限,可以選擇比我這個更新的版本;py3.6表示自己的python版本,必須對應;cuda92表示我的cuda版本號,7_0是cudnn的版本號。這兩者也必須和自己所安裝的版本對應; -
下載好上面那個torch包后,在Anaconda prompt 里利用
cd命令定位到自己下載的文件所在位置,在base環境下安裝就好,不需要新建envs; 如下圖1所示;因為是壓縮包,所以安裝命令為下下圖2所示,安裝很快,包括下一步的安裝,所以才極力推薦這種離線安裝版本:

conda install --use-local ****.bz2 //****部分為自己下載的文件名
- 還是去該網址 下載
torchvision-0.5.0-py36_cu92.tar.bz2,除了py36和cu92與自己的版本號對應之外,其余的數字號可以自己選擇。下載完成后,在Anaconda prompt里,利用cd命令定位到自己下載的文件所在的位置,然后輸入如下代碼。等待安裝完畢
conda install -use-local torchvision-0.5.0-py36_cu92.tar.bz2
- 輸入以下代碼檢驗
> python
> import torch
> print(torch.cuda.is_available())
> # 如果顯示結果為 True, 則表示全程安裝成功。
