From 6140f612939b9abc25bd169baed7c1f85a7b77d6 Mon Sep 17 00:00:00 2001 From: Jan Siwiec <jan.siwiec@vsb.cz> Date: Tue, 18 Oct 2022 09:36:06 +0200 Subject: [PATCH] Update tensorflow.md --- docs.it4i/software/machine-learning/tensorflow.md | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/docs.it4i/software/machine-learning/tensorflow.md b/docs.it4i/software/machine-learning/tensorflow.md index 9fc9a1ab0..8e7386269 100644 --- a/docs.it4i/software/machine-learning/tensorflow.md +++ b/docs.it4i/software/machine-learning/tensorflow.md @@ -4,14 +4,16 @@ TensorFlow (TF) is an open-source software library which can compile tensor oper very quickly on both CPUs and GPUs. It is often used as a backend for machine learning libraries and models. -We heavily recommend the usage of `Tensorflow 2.x`. TensorFlow 1 has been long deprecated and it +We heavily recommend the usage of `TensorFlow 2.x`. TensorFlow 1 has been long deprecated and it will probably be difficult to make it run on GPUs on our clusters. ## Installation + For TensorFlow to work with GPUs, you have to use several libraries (CUDA, cuDNN, NCCL etc.) with versions that are compatible together. You can load the correct modules with the following command: + ```console $ ml TensorFlow ``` @@ -19,6 +21,7 @@ $ ml TensorFlow If you want to upgrade the TensorFlow version used in this package or install additional Python modules, you can simply create a virtual environment and install a different TensorFlow version inside it: + ```console $ python3 -m venv venv $ source venv/bin/activate @@ -53,7 +56,7 @@ Using the available `TensorFlow` module should make sure that these modules will ### Selecting GPUs -You can select how many and which (Nvidia) GPUs will be used by TensorFlow with the +You can select how many and which (NVIDIA) GPUs will be used by TensorFlow with the `CUDA_VISIBLE_DEVICES` environment variable. ```console -- GitLab