Skip to content

pyhf Docker images built on Nvidia Container Toolkit enabled base images

License

Notifications You must be signed in to change notification settings

pyhf/cuda-images

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pyhf NVIDIA CUDA enabled Docker images

pyhf Docker images built on the NVIDIA CUDA enabled images for runtime use with the the NVIDIA Container Toolkit.

These images are meant to be used as base images for bespoke user application images to build on.

Docker Images Docker Pulls

Installation

Example:

docker pull pyhf/cuda:0.6.3-jax-cuda-11.1

Use

To check that NVIDIA GPUS are being properly detected run

docker run --rm --gpus all pyhf/cuda:0.6.3-jax-cuda-11.1 'nvidia-smi'

and check if the nvidia-smi output appears correctly.

To run (interactively) using GPUs on the host machine:

docker run --rm -ti --gpus all pyhf/cuda:0.6.3-jax-cuda-11.1

Tests

To verify things are working on your host machine you can run

docker run --rm --gpus all -v $PWD:$PWD -w $PWD pyhf/cuda:0.6.3-jax-cuda-11.1 'bash tests/test_backend.sh jax'