Main Navigation

Secondary Navigation

Page Contents

Contents

Pytorch and Tensorflow

We have installed pytorch and tensorflow in numerous versions. These packages are optimized for our (Nvidia) GPUs and therefore strongly recommended. You can use them in a so called singularity container. To get an overview over installed packages, please use
rhrk-singularity --list
You may work and test directly within an image or you may encapsulate your programs in a script to be submitted via sbatch like:
#!/bin/bash
#SBATCH --gres=gpu:V100:1 # select a host with a Volta GPU

rhrk-singularity tensorflow_21.06-tf1-py3.simg python3 your_python_file.py