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

cuda 版本更新_第1张图片
  1. 选择完后,复制下方链接到Linux终端

cuda 版本更新_第2张图片

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. 安装过程中会弹出如下界面

cuda 版本更新_第3张图片

由于之前已经安装过显卡驱动, 这里选中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

cuda 版本更新_第4张图片
cuda 版本更新_第5张图片
  1. 保存后退出输入 source ~/.bashrc 激活

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

cuda 版本更新_第6张图片
  1. 重新下载对应cuda11.1 版本的pytorch即可

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

你可能感兴趣的:(pytorch,人工智能,python)