0 准备
在终端输入如下两个命令,确保已经安装显卡驱动以及docker
nvcc --version
docker
1 设置stable存储库和密钥
进入NVIDIA/nvidia-docker github
点击进入“安装向导”,依次输入如下代码,即可安装成功。
测试
输入
sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi
显示如下图,代表安装成功。
在终端输入如下两个命令,确保已经安装显卡驱动以及docker
nvcc --version
docker
进入NVIDIA/nvidia-docker github
点击进入“安装向导”,依次输入如下代码,即可安装成功。
输入
sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi
显示如下图,代表安装成功。