安裝gpu版本的pytorch須要三個東西:pytorch(torchvision)、cuda、cudnnwindows
相信你們都安裝過了anaconda,就不介紹anaconda的安裝了ui
1.安裝cuda:從官網下載cuda,這裏我使用的是cuda 9.0版本(10.0的不知道出於什麼緣由沒成功使用)url
下載以後就是一鍵式操做,一直點next就行了spa
2.安裝cudnn:cudnn可以進一步加速速度。所以推薦安裝。.net
從官網下載cudnn,可是要註冊一個帳號(很簡單的註冊)【注意cudnn與cuda有版本間的對應關係 對應cuda9.0版本的話使用的是7.4】code
下載以後解壓並複製相應文件夾的內容到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0blog
添加環境變量:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64教程
3.更改鏡像:由於官網的速度實在太慢太慢了get
添加清華鏡像
1 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ 2 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ 3 conda config --set show_channel_urls yes
添加pytorch鏡像
1 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
4.安裝pytorch:從官網得到conda指令 :conda install pytorch torchvision cudatoolkit=9.0 由於修改了鏡像,因此去掉了官網的給的地址(-c pytorch)it
5.驗證是否安裝成功:
1 import torch 2 torch.cuda.is_available()
若是是True恭喜你啦!
6.感謝我以前看過的教程: