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
相关文章
相关标签/搜索