Ubuntu20.04安装CUDA

1、安装NVIDIA驱动

通过以下命令检查驱动安装情况

nvidia-smi

2、安装CUDA

首先Ubuntu 20.04默认g++9版本太高，会导致CUDA无法安装，因此要先降低g++版本。

sudo apt-get install gcc-7 g++-7
sudo update-alternatives

用runfile安装，两步：

wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run
sudo sh cuda_11.1.1_455.32.00_linux.run

这部分参考文章：Ubuntu 20.04安装CUDA & CUDNN 手把手带你撸_哈希Map的博客-CSDN博客_ubuntu20.04安装cuda

3、配置CUDA的环境变量

在终端输入：

sudo gedit ~/.bashrc

在最后一行添加：

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64
export PATH=$PATH:/usr/local/cuda-11.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.1

注意：这里为安装的CUDA版本是11.1，按照自己的更改

4、测试CUDA是否安装成功

这部分参考文章：Ubuntu20.04下CUDA、cuDNN的详细安装与配置过程（图文）_嵌入式技术的博客-CSDN博客_ubuntu 安装cudnn