GPU版PyTorch Windows 2023年下载安装教程

基本教程

下载安装GPU版的Pytorch网络上已经有很多教程,Anaconda+PyTorch这里推荐b站的我是土堆up主的安装教程视频,简单易懂,跟着一步步走大体上是没有问题的

但!!!
视频是2022年的,2023年pytorch更新版本后有些许的不一样,下面分享下我这次下载安装遇到的问题和解决办法

问题:python-cuda镜像源找不到文件

按照官网指令下载网络太慢了,所以我选择镜像源下载,这里我选的北京外国语大学的镜像网站

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/win-64/

但是终端显示有packages conflicts,类似这样

Specifications:
 
  - torchaudio -> python[version='2.7.*|3.5.*|3.6.*|>=2.7,<2.8.0a0|>=3.5,<3.6.0a0|3.4.*']
 
Your python: python=3.9

解决办法

pytorch更新后cudatooklist改成了pytorch-cuda,要在nvidia官方网站下载,镜像网站没有及时更新,所以要在后面加上nvidia官网下载源,输入下面的指令就可以找到正确的版本

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/win-64/ -c nvidia

或者先在channels里添加好镜像网站,再直接输入下面的指令也可以成功

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c nvidia

这样系统在镜像网站找不到的包就可以去nvidia搜索下载

问题:镜像网站只推送cpu版本的pytorch

这个问题搜索了很多解决办法,最后解决的办法参考了这篇博文,就是需要先在终端输入官网的指令

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

等待一会儿看要下载的pytorch,torchvision和torchaudio的版本,我这里是pytorch-2.0.1, torchvision-0.15.2, torchaudio-2.0.2, cuda-11.8
所以我输入下面的指令就可以了,但镜像网站还是偶尔会发疯,大家多尝试几次说不定就好了

conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c nvidia

问题:下载过程中HTTP出错

这个重新再输入一遍一样的指令,或者关了终端重新打开,多尝试几次应该就可以,就是一个连接的问题


这些就大概是我遇到的问题和解决办法,其实配置好镜像源后安装速度挺快的,也很顺畅,主要就是注意一个Anaconda和cuda的版本问题,Anaconda建议选2022年的,毕竟现在已经2023年了,下载安装之前也更新一下自己的conda,版本太老也会出现些不必要的麻烦

还要注意有时候用镜像源安装的包可能会少,在下载之前检查一下所需内存和官网下载所需内存是不是一样的

最后祝大家都能顺利下载安装!

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值