不同的pytorch,torchvision版本导致的key error的错误

本文介绍了解决PyTorch和torchvision版本冲突的方法,包括从清华源下载相应版本并手动上传到服务器安装,以及如何处理因CUDA版本不匹配导致的错误。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

如题,解决办法

只需要将原来版本的pytorch,torchvision卸载即可

conda list中找到pytorch,torchvision

把它们卸载,即 conda uninstall pytorch torchvision

我选择的做法是从清华源上直接下载相应版本上传到服务器自己手动安装

清华源为https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

找到相应torchvision,pytorch版本号,下载tar.bz2的文件 

然后再利用 conda install --use-local  ***.tar.bz2  安装相应的pytorch,torchvision包

此时,如果import torch可能会报错,可能是因为cuda版本不同导致的

不用着急搜如何卸载cuda再安装, 直接输入安装相应版本的cuda命令就会自动替换了

命令google一下就好 比如google "conda install cuda 8" 就可以了

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值