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。