Skip to content

Argonne Leadership Computing Facility

Data Science Software Availability

On ThetaGPU, currently we support the major deep learning frameworks through two paths: Singularity containers, based off of NVIDIA's Docker containers, and through bare-metal source builds. The bare-metal builds are for TensorFlow 2.X PyTorch. TensorFlow 1.X is supported only via NVIDIA's containers at this time.


As of now, the NVIDIA containers with TensorFlow 1, 2 and PyTorch built against cuda11, cudnn8 are available in singularity format here:

$ ls /lus/theta-fs0/projects/datascience/thetaGPU/containers/ 
pytorch_20.08-py3.sif tf1_20.08-py3.sif tf2_20.08-py3.sif

Execute a container interactively like this:

$ singularity exec --nv -B /lus:/lus /lus/theta-fs0/projects/datascience/thetaGPU/containers/tf1_20.08-py3.sif bash