Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

https://github.com/pyhf/cuda-images

pyhf Docker images built on Nvidia Container Toolkit enabled base images
https://github.com/pyhf/cuda-images

cuda jax nvidia nvidia-cuda nvidia-docker pyhf

Last synced: 4 months ago
JSON representation

pyhf Docker images built on Nvidia Container Toolkit enabled base images

Host: GitHub
URL: https://github.com/pyhf/cuda-images
Owner: pyhf
License: apache-2.0
Created: 2021-06-01T06:10:56.000Z (about 3 years ago)
Default Branch: main
Last Pushed: 2021-10-24T22:50:30.000Z (over 2 years ago)
Last Synced: 2024-01-16T10:47:07.360Z (5 months ago)
Topics: cuda, jax, nvidia, nvidia-cuda, nvidia-docker, pyhf
Language: Shell
Homepage: https://hub.docker.com/r/pyhf/cuda/
Size: 19.5 KB
Stars: 5
Watchers: 4
Forks: 0
Open Issues: 1
Metadata Files:
- Readme: README.md
- License: LICENSE

Lists

awesome-stars - pyhf/cuda-images - pyhf Docker images built on Nvidia Container Toolkit enabled base images (Shell)

README

# `pyhf` NVIDIA CUDA enabled Docker images

[`pyhf`](https://pyhf.readthedocs.io/) Docker images built on the [NVIDIA CUDA enabled images](https://github.com/NVIDIA/nvidia-docker) 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](https://github.com/pyhf/cuda-images/actions/workflows/docker.yml/badge.svg?branch=main)](https://github.com/pyhf/cuda-images/actions/workflows/docker.yml?query=branch%3Amain)
[![Docker Pulls](https://img.shields.io/docker/pulls/pyhf/cuda.svg)](https://hub.docker.com/r/pyhf/cuda/)

## Installation

- Make sure that you have the [`nvidia-container-toolkit`](https://github.com/NVIDIA/nvidia-docker) installed on the host machine
- Check the [list of available tags on Docker Hub](https://hub.docker.com/r/pyhf/cuda/tags?page=1) to find the tag you want
- Use `docker pull` to pull down the image corresponding to the tag

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`](https://developer.nvidia.com/nvidia-system-management-interface) 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'
```