Linux服务器升级pytorch版本踩坑

背景

先说下背景：本地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
安装完毕以后再运行就可以了！