diff --git a/docs.it4i/software/machine-learning/tensorflow.md b/docs.it4i/software/machine-learning/tensorflow.md index 9fc9a1ab0cd7062270e9f30055e3c00cca199d0a..8e7386269a8c778331b87eb40d1d45da22adc678 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