优雅的深度学习搭建

以下以centos 7 为 安装常用的软件 可跳过

yum install zsh vim git sh -c "$(curl -fsSL https://raw.github.com/robbyrussell/oh-my-zsh/master/tools/install.sh)"

安装docker和 nvidia-docker
curl -fsSL https://get.docker.com -o get-docker.sh chmod 777 get-docker.sh ./get-docker.sh distribution=$(. /etc/os-release; echo $ID$VERSION_ID) curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | \ sudo tee /etc/yum.repos.d/nvidia-docker.repo sudo yum install -y nvidia-docker2 sudo pkill -SIGHUP dockerd

安装nvidia驱动,重启系统
sudo yum install nvidia-x11-drv sudo reboot

安装tensorflow-gpu-py3-jupyter镜像
docker run --runtime=nvidia -it-p 8888:8888-v /home:/hometensorflow/tensorflow:latest-gpu-py3-jupyter

打完收工

    推荐阅读