CUDA、Docker、Docker Compose、Nvidia-Docker的设置备忘录

设置CUDA、Docker、Docker Compose和Nvidia-Docker的备忘录。

① 2018年4月上旬,在Ubuntu 16.04操作系统中,使用GeForce GTX TITAN X显卡和CUDA 9.1。

CUDA是一款用于并行计算的计算机编程模型和平台。

从这里(https://developer.nvidia.com/cuda-downloads)下载文件,并按照上面的步骤进行安装。这个地方是绝对不会失败的。

$ wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.1.85-1_amd64.deb
$ dpkg -i cuda-repo-ubuntu1604_9.1.85-1_amd64.deb
$ apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
$ apt-get update
$ apt-get install cuda

司机

我想从这里(http://www.nvidia.com/Download/index.aspx)下载,但是选择TITAN X,Ubuntu 16.04后却发现不存在。因此,我将按照以下步骤安装驱动程序。

$ add-apt-repository ppa:graphics-drivers/ppa
$ apt-get update
$ apt-get install nvidia-390

核实操作:nvidia-smi

Docker:使用 Docker

$ wget -qO- https://get.docker.com/ | sh

动作确认: 使用docker run –rm hello-world命令

由于无法安装nvidia-docker,所以我进行了降级(从18.04.0降到18.03.0)。

$ apt-get install docker-ce=18.03.0~ce-0~ubuntu

看起来我误会了,似乎不需要预先安装Docker,只需要安装nvidia-docker就可以了。

英伟达容器 dá qì)

只要按照上述方法降级,就可以使用apt-get进行安装。不需要事先安装docker,似乎可以使用以下命令进行安装,其中还包括docker。

使用 https://nvidia.github.io/nvidia-docker/ 作为指南,将存储库添加进去。

$ curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -
$ distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
$ curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list
$ sudo apt-get update

然后安装。

$ apt-get install nvidia-docker2
$ pkill -SIGHUP dockerd

确认动作:nvidia-docker运行命令 –rm nvidia/cuda:9.1-cudnn7-devel-ubuntu16.04 nvidia-smi

docker-compose: Docker容器编排工具

$ pip install docker-compose

2018年5月上旬,使用的操作系统为Ubuntu 18.04,显卡为Quadro K2200,CUDA版本为9.1。

CUDA 加速技术

目前还不支持18.04。(但很可能很快会支持)
所以,我使用官方提供的17.04用的Runfile进行了安装。

$ apt-get install g++-6
$ apt-get install nvidia-cuda-toolkit

$ wget https://developer.nvidia.com/compute/cuda/9.1/Prod/local_installers/cuda_9.1.85_387.26_linux
$ ./cuda_9.1.85_387.26_linux --override

在安装过程中会询问”是否要在不受支持的系统上安装?”,我会选择”是”来回答。接下来会问”是否要安装驱动程序?”,但由于稍后会安装,我会选择”否”来回答。然后随便回答就可以继续安装了。

司机

$ apt-get install nvidia-driver-390

NVIDIA-Docker

在不需要预先安装Docker的情况下,可以通过步骤①安装nvidia-docker2。

广告
将在 10 秒后关闭
bannerAds