WIN10中Anaconda安装gpu版本的pytorch

在WIN10系统下,经历多次尝试,成功安装了GPU版本的PyTorch。首先确定CUDA版本,然后在conda环境中创建并激活PyTorch环境,修改conda镜像源为清华源,根据CUDA版本选择合适的PyTorch版本进行安装。在安装过程中遇到了CondaHTTPError和下载超时问题,通过更换HTTP源和调整超时时间解决。最终成功安装PyTorch。
摘要由CSDN通过智能技术生成

WIN10环境下安装pytorch:

尝试安装gpu版本的pytorch,尝试无数次以后,终于成功。

安装步骤:

1.首先查找自己gpu版本信息,在cmd中输入:

nvidia-smi

得到版本信息
可以看到CUDA Version我这里的CUDA支持11.1,在去pytorch官网看到可以安装10.1版本。
2.conda prompt中创建pytorch环境。

conda create --name torchenv python=3.6 

3.激活创建的环境

activate torchenv

4.改掉conda安装的镜像源,将以下复制进.condarc文件中。

  channels:
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值