Ubuntu14.04安装nvidia-docker2

时间:2023-03-09 17:14:32
Ubuntu14.04安装nvidia-docker2

1、在安装nvidia-docker2以前需要先安装docker:

首先添加PPA源并更新源:
add-apt-repository ppa:ubuntu-sdk-team/ppa
apt-get update
安装docker18.03版本:
sudo apt-get install docker-ce=18.03.~ce-~ubuntu

2、安装nvidia-docker2:

sudo apt-get install nvidia-docker2=2.0.+docker18.03.1- nvidia-container-runtime=2.0.+docker18.03.1-

3、测试nvidia-docker2:

由于本人电脑的CUDA版本为8.0.6,因此这里需要指定CUDA版本为8.
docker run --runtime=nvidia --rm nvidia/cuda:8.0-devel nvidia-smi