Skip to content

Instantly share code, notes, and snippets.

@sanketdiwale
Last active May 7, 2022 22:03
Show Gist options
  • Save sanketdiwale/06a4ecbfadf076c9c885634754d3f889 to your computer and use it in GitHub Desktop.
Save sanketdiwale/06a4ecbfadf076c9c885634754d3f889 to your computer and use it in GitHub Desktop.
How to install tensorflow-gpu on Engaging clusters
module load python/3.9.4
module load cuda/11.3
module load cudnn/8.2.1_cuda11.3
module load gcc/9.3.0
module load cmake/3.17.3
pip install --user cython
pip install --user --upgrade tensorflow-gpu
pip install --user --upgrade tensorflow-probability
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment