docker run --gpus all nvidia/cuda:10.0-base nvidia-smi docker run --gpus 2 nvidia/cuda:10.0-base nvidia-smi docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi
docker run -it --rm tensorflow/tensorflow:1.15.2-gpu-py3 \ python -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.reduce_sum(tf.random_normal([1000, 1000])))"
Tensorflow 2.X
1 2 3 4
docker run -it --gpus all tensorflow/tensorflow:2.0.1-gpu-py3\ python -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"#OK
docker run --gpus all -it tensorflow/tensorflow:2.0.1-gpu-py3 bash #OK