安装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.感谢我以前看过的教程: