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

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

本文在介绍pytorch前默认安装了Anaconda,cuda,cudnn
同时需要清楚自己的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

  • 4
    点赞
  • 25
    收藏
    觉得还不错? 一键收藏
  • 8
    评论
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值