win10安装pytorch

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

相关文章
相关标签/搜索