注意: 不要采用之前安装在UBUNT18.04时 所用的sudo ubuntu-drivers autoinstall 命令自动安装驱动,安装后了造成内核崩溃, WIFI 适配器也无法识别!
安装CUDA 三件套:
1.NVIDIA GPU 驱动 (我的是GTX 1060)
2.Cuda toolkit 11.2
3.cuDNN (给DNN用的驱动)
$ sudo apt-get install linux-headers-$(uname -r)
$ distribution=$(. /etc/os-release;echo $ID$VERSION_ID | sed -e 's/\.//g')
$ wget https://developer.download.nvidia.com/compute/cuda/repos/$distribution/x86_64/cuda-keyring_1.0-1_all.deb
$ sudo dpkg -i cuda-keyring_1.0-1_all.deb
$ sudo apt-get update
$ sudo apt-get -y install cuda-drivers
安装后检查:
$ nvidia-smi
2. 安装CUDA Toolkit 11.2, 采用方法是 install CUDA toolkit from CUDA repository。
先点击这里的官网,下载
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda-repo-ubuntu2004-11-2-local_11.2.0-460.27.04-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-2-local_11.2.0-460.27.04-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-2-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
使用命令
$ nvcc -V
检查是否安装成功, 如果没成功, 用以下命令添加环境变量即可:
echo 'export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}' >> ~/.bashrc
source .bashrc
以下是示列:
note: 估计官方的命令错误, 安装时命令显示的是cuda 11.2版本, 装好后成了11.8的版本。