Ubuntu14.04安裝nvidia-docker2

一、在安裝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
相關文章
相關標籤/搜索