win10安装pytorch
安装gpu版本的pytorch需要三个东西:pytorch(torchvision)、cuda、cudnn
相信大家都安装过了anaconda,就不介绍anaconda的安装了
1.安装cuda:从官网下载cuda,这里我使用的是cuda 9.0版本(10.0的不知道出于什么原因没成功使用)
下载之后就是一键式操作,一直点next就好了
2.安装cudnn:cudnn能够进一步加速速度。因此推荐安装。
从官网下载cudnn,但是要注册一个账号(很简单的注册)【注意cudnn与cuda有版本间的对应关系 对应cuda9.0版本的话使用的是7.4】
下载之后解压并复制相应文件夹的内容到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
添加环境变量:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
3.更改镜像:因为官网的速度实在太慢太慢了
添加清华镜像
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)
5.验证是否安装成功:
1 import torch
2 torch.cuda.is_available()
如果是True恭喜你啦!
6.感谢我之前看过的教程: