一、在安裝nvidia-docker2之前須要先安裝docker:docker
首先添加PPA源並更新源: add-apt-repository ppa:ubuntu-sdk-team/ppa apt-get update 安裝docker18.03版本: sudo apt-get install docker-ce=18.03.1~ce-0~ubuntu
二、安裝nvidia-docker2:ubuntu
sudo apt-get install nvidia-docker2=2.0.3+docker18.03.1-1 nvidia-container-runtime=2.0.0+docker18.03.1-1
三、測試nvidia-docker2:測試
因爲本人電腦的CUDA版本爲8.0.6,所以這裏須要指定CUDA版本爲8.0 docker run --runtime=nvidia --rm nvidia/cuda:8.0-devel nvidia-smi