GPU版本Pytorch环境配置

0 写在前面

最近因为代码需要,要求 P y t h o n ⩾ 3.8.0 Python \geqslant 3.8.0 Python3.8.0,因此在主机上重新安装了3.10.11版本的(虽然anaconda有着出色的环境管理,但本人还是习惯按需自行配置,故没有安装anaconda),在离线版本安装包中找到跟已装CUDA(11.7)版本对应的文件 torch-1.13.0+cu117-cp310-cp310-win_amd64.whl 下载,且能够顺利安装。但是在之后安装配件 torchvision 和 torchaudio 发现离线安装时会导致第一步安装的gpu-torch先被卸载,安装cpu版本的之后,才能安装上 torchvision/torchaudio 。

有点莫名其妙。torchvision/torchaudio 下载 pip install的 都是gpu版本。
torchvision-0.14.1+cu117-cp310-cp310-win_amd64.whl
torchvision-0.15.1+cu117-cp310-cp310-win_amd64.whl

1 解决

看见一个博主安装的如下两个特定版本,尝试下,就顺利了,也是蛮奇怪的,
torchaudio-0.13.0+cu117-cp310-cp310-win_amd64.whl
torchvision-0.14.0+cu117-cp310-cp310-win_amd64.whl

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值