Add pytorch packages and update fidle env test

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)

docker/requirements.txt

@@ -5,4 +5,7 @@ Matplotlib
 Pandas
 Pandoc
 pyyaml
+torch
+torchvision
+torchaudio
 Jupyterlab