linux 安装对应版本的cuba和pytorch

查看显卡驱动，cuda版本

nvidia-smi

找不到对应的cuda的pytorch，cuda版本过高，需要卸载显卡驱动，sodo apt remove --auto-remove nvidia-driver-440
然后查看服务器适合的驱动ubuntu-drivers devices
然后安装 sudo apt install dia-driver-450-server
然后重启 reboot

查看cuba版本

cat /usr/local/cuda/version.txt
或者
nvcc -v

到官网安装对应版本的pytorch

https://pytorch.org/get-started/locally/#mac-installation
可以安装老版本的

install previous versions of PyTorch

推荐使用conda安装方便版本控制

pytorch下载过慢

使用清华镜像源

然后去掉 -c pytorch