From 2de86895e0df486a5212fbead46882bd4683ba18 Mon Sep 17 00:00:00 2001
From: Soraya Arias <soraya.arias@inria.fr>
Date: Fri, 11 Mar 2022 12:17:55 +0100
Subject: [PATCH] Add pytorch packages and update fidle env test
---
docker/fidle_env_test.py | 8 ++++++++
docker/requirements.txt | 3 +++
2 files changed, 11 insertions(+)
diff --git a/docker/fidle_env_test.py b/docker/fidle_env_test.py
index f3b631d..592e6eb 100644
--- a/docker/fidle_env_test.py
+++ b/docker/fidle_env_test.py
@@ -3,6 +3,7 @@
##
import tensorflow as tf
+import torch
import sys, os
# Check data set is found
@@ -16,5 +17,12 @@ print("FIDLE_DATASETS_DIR = ", os.path.expanduser(datasets_dir))
print("Python version = {}.{}".format(sys.version_info[0], sys.version_info[1]))
# Check tensorflow version
print("Tensorflow version = ", tf.__version__)
+# Obsolete command
+#print("Tensorflow GPU/CUDA available = ", tf.test.is_gpu_available())
+print("Tensorflow GPU/CUDA available = ", "true" if len(tf.config.list_physical_devices('GPU')) else "False")
+
+# Chech Pytorch version
+print("Pytorch version = ", torch.__version__)
+print("Pytorch GPU/CUDA available = ", torch.cuda.is_available())
sys.exit(0)
diff --git a/docker/requirements.txt b/docker/requirements.txt
index 39ea851..0118e42 100644
--- a/docker/requirements.txt
+++ b/docker/requirements.txt
@@ -5,4 +5,7 @@ Matplotlib
Pandas
Pandoc
pyyaml
+torch
+torchvision
+torchaudio
Jupyterlab
--
GitLab