docker gpu 创建 训练环境_Docker GPU

插播一个备忘录

sudo apt install docker.io

sudo usermod -a -G docker lizhe

然后正常退出，重启docker保证能正常使用

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -

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

sudo apt-get install -y nvidia-container-toolkit

sudo systemctl restart docker

docker run --gpus all nvidia/cuda:10.0-base nvidia-smi

docker run --gpus all --name tensorflow1 -it -p 8080:8888 tensorflow/tensorflow:latest-gpu-py3-jupyter

import tensorflow as tf

import os

os.environ['TF_CPP_MIN_LOG_LEVEL']='2'

a = tf.constant(1.)

b = tf.constant(2.)

print(a+b)

print('GPU:', tf.test.is_gpu_available())