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的博客-优快云博客_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的详细安装与配置过程(图文)_嵌入式技术的博客-优快云博客_ubuntu 安装cudnn
本文详细介绍了如何在Ubuntu20.04上安装NVIDIA驱动和CUDA。首先通过`nvidia-smi`检查驱动状态,然后降低g++版本以解决CUDA安装问题,接着使用runfile安装CUDA 11.1,并更新环境变量。最后通过运行测试程序验证CUDA安装成功。
1万+

被折叠的 条评论
为什么被折叠?



