cuda 版本更新

RTX 3090 算力为 8.6，pytorch 版本要11.1以上,不然无法让代码在GPU上跑，对应的cuda版本也要更新

一，卸载原有的cuda

sudo /usr/local/cuda-10.0/bin/uninstall_cuda_10.0.pl
sudo rm -rf /usr/local/cuda-10.0

二，安装cuda

1. cuda11.1网址：CUDA Toolkit 11.1.0 | NVIDIA Developer

1. 选择完后，复制下方链接到Linux终端

wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run

sudo sh cuda_11.1.0_455.23.05_linux.run

1. 安装过程中会弹出如下界面

由于之前已经安装过显卡驱动， 这里选中driver，点击回车将X号去掉， 然后选择Install回车即可正常安装 4. 打开.bashrc文件（在home目录），输入以下几行

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

export PATH=$PATH:/usr/local/cuda/bin

export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

1. 保存后退出输入 source ~/.bashrc 激活

2. 在终端中输入 nvcc -V查看cuda是否安装成功

1. 重新下载对应cuda11.1 版本的pytorch即可

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c nvidia