From 2e8ea7040a2d76ae17d8f051bb09af8d25b674b6 Mon Sep 17 00:00:00 2001 From: "Gerber, Mike" Date: Tue, 25 Feb 2020 18:14:03 +0100 Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20tensorflow-vs-tensorflow-gpu?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tensorflow-vs-tensorflow-gpu | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100755 tensorflow-vs-tensorflow-gpu diff --git a/tensorflow-vs-tensorflow-gpu b/tensorflow-vs-tensorflow-gpu new file mode 100755 index 0000000..a2171ba --- /dev/null +++ b/tensorflow-vs-tensorflow-gpu @@ -0,0 +1,19 @@ +#!/bin/sh +for package in "tensorflow==1.15.*" "tensorflow-gpu==1.15.*"; do + for CUDA_VISIBLE_DEVICES in "0" ""; do + + echo "== $package, CUDA_VISIBLE_DEVICES='$CUDA_VISIBLE_DEVICES'" + + export CUDA_VISIBLE_DEVICES + + venv=/tmp/tmp.$RANDOM + virtualenv --quiet -p /usr/bin/python3 $venv + . $venv/bin/activate + + pip3 install --quiet --upgrade pip + pip3 install --quiet "$package" + + python3 -c 'import tensorflow as tf; print("GPU available:", tf.test.is_gpu_available())' + + done +done