From ee7f6593d545c1dc0c634f57c5f80cfaff6db7c9 Mon Sep 17 00:00:00 2001 From: "Gerber, Mike" Date: Tue, 15 Oct 2019 12:46:51 +0200 Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20Initial=20commit?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- Dockerfile | 10 ++++++++++ requirements.txt | 1 + run | 5 +++++ run-docker | 3 +++ test-nvidia | 13 +++++++++++++ 5 files changed, 32 insertions(+) create mode 100644 Dockerfile create mode 100644 requirements.txt create mode 100755 run create mode 100755 run-docker create mode 100755 test-nvidia diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..94728a1 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,10 @@ +FROM nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04 +RUN apt-get update &&\ + apt-get install -y python3 python3-pip &&\ + apt-get clean && rm -rf /var/lib/apt/lists/* +COPY requirements.txt /tmp +RUN pip3 install -r /tmp/requirements.txt + +COPY test-nvidia /usr/bin + +CMD ["/usr/bin/test-nvidia"] diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..c21b1c7 --- /dev/null +++ b/requirements.txt @@ -0,0 +1 @@ +tensorflow-gpu == 1.14.0 diff --git a/run b/run new file mode 100755 index 0000000..691ef38 --- /dev/null +++ b/run @@ -0,0 +1,5 @@ +#!/bin/sh +virtualenv -p /usr/bin/python3 ~/.virtualenvs/test-nvidia +. ~/.virtualenvs/test-nvidia/bin/activate +pip3 install -r requirements.txt +python3 test-nvidia diff --git a/run-docker b/run-docker new file mode 100755 index 0000000..1002148 --- /dev/null +++ b/run-docker @@ -0,0 +1,3 @@ +#!/bin/sh +docker build -t test-nvidia `dirname $0` +docker run --gpus all -it --rm test-nvidia diff --git a/test-nvidia b/test-nvidia new file mode 100755 index 0000000..b3a2751 --- /dev/null +++ b/test-nvidia @@ -0,0 +1,13 @@ +#!/usr/bin/python3 +import os +import tensorflow as tf + + +os.system('nvidia-smi') + + +hello = tf.constant('Hello, TensorFlow!') +sess = tf.compat.v1.Session() +print(sess.run(hello)) +print('GPU available:', tf.test.is_gpu_available(cuda_only=True)) +