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

posted @ 2019-02-12 19:09  suntor  阅读(4386)  评论(1编辑  收藏  举报