【亲测】ubuntu20.4 安装CUDA11.8

首先回答以下为什么选择安装cuda11.8，原因是目前pytorch架构最高支持11.8版本，所以为了更好的使用pytorch,我安装cuda11.8.

一、在NVIDIA官网下载CUDA11.8

1、网址：CUDA Toolkit Archive | NVIDIA Developer

2、选择CUDA11.8

3、选择安装系统信息

4、这里Installer Type我们选择deb(local)方式安装（下载速度快.......），此时会出现安装步骤：

5、以此按上面的提示进行安装即可(安装过程中有点耗时、有点卡，请耐心等待)

二、配置环境

1、sudo gedit ~/.bashrc

2、在打开的 .bashrc最后添加

export PATH=$PATH:/usr/local/cuda-11.8/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.8/lib64

3、配置环境生效

source ~/.bashrc

三、查看CUDA安装信息

1、查看版本号

nvcc --version

此刻CUDA11.8安装成功！

四、卸载CUDA11.8

1、sudo apt-get --purge remove cuda*

2、sudo apt-get autoremove

3、sudo reboot