Merge pull request #77 from carlos-encs/tensorf-cuda

Add Tensorflow - Conda/CUDA example

Author: carlos-encs

src/tensorflow-conda/README.md

@@ -0,0 +1,23 @@
+# Tensorflow in Conda
+Installation and configuration of a Conda environment for tensorflow with Cuda compatibility
+
+## Instructions
+* Conda environment creation, tensorflow and cuda packages installation
+
+    * Connect to a node (salloc):
+```bash
+mkdir -p /speed-scratch/$USER/envs/{tmp,pkgs,cache}
+setenv ENV_DIR /speed-scratch/$USER/envs
+setenv CONDA_PKGS_DIRS /speed-scratch/$USER/envs/pkgs
+conda create -y -p $ENV_DIR/conda-env python=3.10
+conda activate $ENV_DIR/conda-env
+conda install -c nvidia cuda-toolkit=12.5
+conda install -c conda-forge tensorflow-gpu matplotlib numpy
+```
+* Modify according your needs (path, Environment_Name) the script `tf-speed.sh`
+* In a submit node, run the script `tf-speed.sh`
+```bash
+sbatch -ppg tf-speed.sh
+```
+The `-p` parameter could have any of our GPU partitions: `pt`, `pg`, `pa`, `cl`, `em`, `pn`
+        

src/tensorflow-conda/tf-speed.sh

@@ -0,0 +1,15 @@
+#!/encs/bin/tcsh
+
+#SBATCH --nodes=1
+#SBATCH --mem=10G
+#SBATCH --gpus=1
+
+set ENV_NAME = conda-env
+set ENV_DIR = /speed-scratch/$USER/envs
+set ENV_PATH = $ENV_DIR/$ENV_NAME 
+setenv CONDA_PKGS_DIRS $ENV_DIR/pkgs 
+# activate the environment created before
+conda activate $ENV_PATH
+# Directory where the script is located
+srun python $PWD/tf-stress.py
+conda deactivate