Linux服务器升级pytorch版本踩坑

本文记录了作者在本地环境中使用PyTorch 1.8.1的torch.fft包时遇到的问题,以及在服务器上由于PyTorch版本不同而导致的缺失该包的情况。文章详细介绍了如何通过升级PyTorch版本并重新安装CUDA工具包来解决这一问题。
部署运行你感兴趣的模型镜像

背景

先说下背景:本地pytorch版本1.8.1,我使用了torch.fft(傅里叶变换)这个包,本地调通了以后打算放到服务器上运行,可是服务器上1.6版本的pytorch没有这个包,我就查了查官方api,里面1.6版本明明显示是有这个包的,如下图所示:
在这里插入图片描述
可是,我导包的时候就是没有,没办法,只能选择更改pytorch的版本。这时候有两个选择:直接升级或者重新新建一个虚拟环境,考虑到新建虚拟环境还得重新下载其他的依赖包,太费事,我就直接在线升级的,使用了pip install --upgrade pytorch torchvision这个命令进行升级,升级完成以后我的pytorch确实变成了1.7.1,但是运行的时候还是报错:找不到cuda,麻蛋!仔细看了看升级的列表,它给我cudatookit10.2给卸载了!!问题是卸载了你再给我安装一个啊!!
导致的问题就是cuda用不了,没办法,只能重装pytorch和cudatoolkit:
conda install pytorch=1.7.1 cudatoolkit=11.0 -c pytorch
安装完毕以后再运行就可以了!

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值