(最新)win10安装pytorch-GPU版本(简洁明了)

​ 上周装了一下tensorflow-GPU版本的,今天来装一下pytorch-GPU版本,有了之前的tensorflow的安装,装pytorch就相对容易了,以下是我边装边写的,尽量简单易懂。

本文在介绍pytorch前默认安装了Anaconda,cuda,cudnn
  • 没装的话可以看我之前的tensorflow的博客
同时需要清楚自己的cuda版本,cudnn版本
我的是:cuda9.0,cudnn7.0.5  系统Win10
cuda版本查看方法: 命令行输入  nvcc -V
  • 因为2019年4月16日和25日清华和中科大分别宣布停止Anaconda镜像服务,这之后装大概率会出现如下错误(网外网速慢):
    在这里插入图片描述

  • 就按下面三条命令来,下载还是很快

# 第一句
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
# 第二句
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
# 第三句
conda config --set show_channel_urls yes
# 第四句
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  • 然后再输安装命令即可,速度真的超级超级超级快的!!!!!
# 9.0是版本号,如果和我不一样的话,记得换一下(-c pytorch这个后缀不要加)
conda install pytorch torchvision cudatoolkit=9.0 
  • 安装成功测试代码
import torch

torch.__version__           # 返回版本号
torch.cuda.is_available()   # 返回True

你可能感兴趣的:(环境配置,安装)