最近几天走了n多弯路 大概做一下总结

1.如果是n卡的话就nvidia-smi 看一下driver版本
根据driver版本再选择cuda版本

(学校的没有公用cuda 装了几次新版才发现这个问题
具体版本的下载地址百度/Google一下都能找到
因为没有sudo 所以要把路径都放到自己的文件夹里 也不要尝试写入自己以外的空间
用nvcc -V检测安装情况

2.装好了cuda之后对照着自己的情况找对应的包
传送门1
有时候下载太慢或者没有对应的版本 可以自己下载了再传过去 pip install xxx.whl
传送门2
一般要装 torch和torchvision 手动找的时候要注意cp python版本和cu cuda版本

import torch
print(torch.cuda.is_available())
#True

返回True就差不多可以告一段落了~