虚拟环境中的torch装错cpu版本,重装gpu版本和对应cuda的解决方法
在虚拟环境中的Cuda和torch不好用
终端服务器里有正常版本的torch和Cuda,但是构建虚拟服务器后跑实验还是cpu版本的,总觉得有问题,但后台的gpu又显示在用,两天都没搞懂咋回事,后面发现是因为师门一起用gpu,我的代码就是只用cpu跑的,所以就重新安装gpu版本。
重新安装torch-gpu版本和Cuda
首先,检查现在安装的torch版本,我当时直接pip install安装的就是cpu版本:
import torch
print(torch.__version__)
这里需要注意,version前后是两个横杠 “__” ,而不是一个横杠 “_” ,有的时候会因为这两条杠中间没有空,而只写一个就会报错。
如果这个的结果输出是cpu版本,就需要先卸载现有的torch,再进行后续的安装:
pip uninstall torch
接下来可以在链接: link查看自己设备对应的版本,比如,我的是需要11.3版本,所以安装口令就是:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu113
这个界面显示的我就是安装好gpu版本的torch和对应的Cuda了。