Skip to content

Commit

Permalink
feat: Update to pyhf v0.7.2
Browse files Browse the repository at this point in the history
  • Loading branch information
matthewfeickert committed May 23, 2023
1 parent e482834 commit cea1732
Show file tree
Hide file tree
Showing 4 changed files with 6 additions and 6 deletions.
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,21 +15,21 @@ These examples assume that you want to use GPU resources to take advantage of ha
Example:

```
docker pull pyhf/cuda:0.7.0-jax-cuda-11.6.0-cudnn8
docker pull pyhf/cuda:0.7.2-jax-cuda-11.6.0-cudnn8
```

### Local use

To check that NVIDIA GPUS are being properly detected run

```
docker run --rm --gpus all pyhf/cuda:0.7.0-jax-cuda-11.6.0-cudnn8 'nvidia-smi'
docker run --rm --gpus all pyhf/cuda:0.7.2-jax-cuda-11.6.0-cudnn8 '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.7.0-jax-cuda-11.6.0-cudnn8
docker run --rm -ti --gpus all pyhf/cuda:0.7.2-jax-cuda-11.6.0-cudnn8
```
2 changes: 1 addition & 1 deletion docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ SHELL [ "/bin/bash", "-c" ]

WORKDIR /home/data

ARG PYHF_VERSION=0.7.0
ARG PYHF_VERSION=0.7.2
ARG PYHF_BACKEND=jax
# Set PATH to pickup virtualenv when it is unpacked
ENV PATH=/usr/local/venv/bin:"${PATH}"
Expand Down
2 changes: 1 addition & 1 deletion htcondor_templates/chtc_hello_gpu/chtc_hello_gpu.sub
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
universe = docker
# To avoid Docker Hub rate limits set as "missing" (default value) unless "always" needed for updated tag
docker_pull_policy = missing
docker_image = pyhf/cuda:0.7.0-jax-cuda-11.6.0-cudnn8
docker_image = pyhf/cuda:0.7.2-jax-cuda-11.6.0-cudnn8

# set the log, error and output files
log = chtc_hello_gpu.log.txt
Expand Down
2 changes: 1 addition & 1 deletion noxfile.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ def build(session):
"nvidia/cuda:11.2.2-devel-ubuntu20.04",
"nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04",
]
pyhf_version = "0.7.0"
pyhf_version = "0.7.2"
pyhf_backend = "jax"
for base_image in target_images:
cuda_version = base_image.split(":")[-1].split("-devel")[0]
Expand Down

0 comments on commit cea1732

Please sign in to comment.