Pytorch1.7.1 GPU版 WIN10安装

博主因安装botorch将PyTorch 1.7.0换成CPU版本的1.7.1,现要换回GPU版本。介绍了安装流程,需进入PyTorch官网,确保CUDA与设备对应,在anaconda prompt中操作。还列举了安装中可能出现的错误,如HTTPError、Conda下载字节不匹配等,并给出解决办法。

0、前言

本来我是有PyTorch1.7.0的,最近想装一个botorch,没注意到他要求torch版本得是1.7.1,就把我的torch1.7.0卸了???然后自动装了一个1.7.1的CPU版本的torch。我记得我当时装PyTorch超级超级麻烦,看了好多博客。现在我得换回GPU版本的,就当是做了一次更新吧。。。

另:本篇写的非常非常简单,我也是参考了别人的,所以在看本篇之前,请先依次阅读

  1. 这篇关于CUDA+CUDNN的安装
  2. 还有这篇关于PyTorch的安装

1、流程

进入PyTorch官网Get Started链接
在这里插入图片描述
CUDA要与自己设备的相对应,查看版本的方法链接
打开anaconda prompt,进入建立好的环境(参见第一节链接)。
我的环境是pytorch_gpu
依次键入、回车

conda activate pytorch_gpu
conda config --set remote_rea
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值