1. 重要的事情
将当前用户添加到docker
用户组
为了避免每次使用docker
命令都需要加上sudo
权限,可以将当前用户加入安装中自动创建的docker
用户组(可以参考官方文档):这是非常重要的一条命令,可以避很多坑!
sudo usermod -aG docker $USER
2. 安装nvidia-docker2
想要在容器使用nvidia-smi,需要安装nvidia-docker2
sudo apt-get install -y nvidia-docker2
测试安装
docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi
docker运行容器时,指定nvidia:
docker run -itd \
--runtime=nvidia --gpus