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
cuda11.1网址：CUDA Toolkit 11.1.0 | NVIDIA Developer
选择完后，复制下方链接到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
安装过程中会弹出如下界面
由于之前已经安装过显卡驱动， 这里选中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
保存后退出输入 source ~/.bashrc 激活
在终端中输入 nvcc -V查看cuda是否安装成功文章来源：https://www.toymoban.com/news/detail-578874.html
重新下载对应cuda11.1 版本的pytorch即可文章来源地址https://www.toymoban.com/news/detail-578874.html
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c nvidia到了这里，关于cuda 版本更新的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
