Ubuntu 上,使用如下命令进行相关环境的配置:

首先,需要先安装 NVIDIA 的驱动,使用如下命令,参考 这里

# list the availabe drivers
sudo ubuntu-drivers list
# choose the default drivers to install
sudo ubuntu-drivers install --gpgpu
# install 
sudo apt install nvidia-utils-550-server
# need reboot 
reboot

此时使用如下命令即可查看是否成功:

nvidia-smi

需要注意的是,此时打印的 CUDA VERSION 并不意味着真的安装了 CUDA ,而只是说明了当前 driver 支持的 CUDA 的最高版本。

也可以使用如下命令来查看已经安装的 driver 的信息:

cat /proc/driver/nvidia/version

然后安装 cuda ,使用如下命令来直接安装最新的 CUDA :

sudo apt install nvidia-cuda-toolkit

在安装结束后,使用如下命令来查看 CUDA 的版本:

nvcc --version