|
这里主要介绍ubuntu20.04英伟达驱动以及cuda的安装教程。 首先是安装NVIDIA驱动，这里采用比较简单的方式，首先查看显卡型号，以及推荐安装驱动版本 #查看型号
ubuntu-drivers devices
如图所示： 这里有两种安装方式： sudo ubuntu-drivers autoinstall
sudo apt install nvidia-driver-495
这样驱动就已经安装完成。
安装cuda11.4 首先从官网现在cuda11.4和cudnn11.4两个文件。 cuda_11.4.0_470.42.01_linux.run cudnn-11.4-linux-x64-v8.2.2.26.tgz
这里由于20.04的gcc版本过高，需要对gcc进行降低版本，才能进行安装，具体操作如下： 这里安装gcc-7 sudo apt-get install gcc-7 g++-7
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 100
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 50
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 100
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-9 50
然后进行安装： 
出入accept 
下一步进行安装cuda，安装时一定不要选择驱动，即无驱动安装，因为上面已经安装过驱动 
安装完成后要进行设置环境变量 export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
也可以还原gcc版本 sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 100
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 50
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-9 100
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 50
安装cudnn8.2 首先进行解压，然后执行如下命令： sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
|
