docker配置gpu开发环境 - Githubissues

zixie1991 / cplusplus-exercise

C++编程练习。。。

8 stars 1 forks source link

docker配置gpu开发环境 #23

Open zixie1991 opened 7 years ago

zixie1991 commented 7 years ago

docker配置gpu开发环境

开发环境为ubuntu 14.04，GPU为N卡

安装docker

参考官方安装文档《Ubuntu安装docker》进行安装

配置docker gpu环境

鉴于NVIDIA提供了docker gpu镜像（Github地址：https://github.com/NVIDIA/nvidia-docker，十分齐全，包括各式cuda版本、cudnn版本），以及详尽的安装方法，所以使用nvidia docker配置环境

安装方法参考了nvidia docker提供的安装说明

# Install nvidia-docker and nvidia-docker-plugin
wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.0-rc.3/nvidia-docker_1.0.0.rc.3-1_amd64.deb
sudo dpkg -i /tmp/nvidia-docker*.deb && rm /tmp/nvidia-docker*.deb

# Test nvidia-smi
nvidia-docker run --rm nvidia/cuda nvidia-smi