Skip to content

Commit

Permalink
Update to CUDA 12.8.0 (#728)
Browse files Browse the repository at this point in the history
This updates to CUDA 12.8.0.

xref: rapidsai/build-planning#139

Authors:
  - Bradley Dice (https://github.com/bdice)

Approvers:
  - Mike Sarahan (https://github.com/msarahan)

URL: #728
  • Loading branch information
bdice authored Feb 3, 2025
1 parent b2701d6 commit 3f8cea7
Show file tree
Hide file tree
Showing 5 changed files with 14 additions and 12 deletions.
2 changes: 2 additions & 0 deletions context/test_notebooks.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,8 @@
'cuspatial/nyc_taxi_years_correlation.ipynb',
# context on skip zipcodes: https://github.com/rapidsai/cuspatial/issues/1426
'cuspatial/ZipCodes_Stops_PiP_cuSpatial.ipynb',
# context on this being skipped: https://github.com/rapidsai/docker/pull/728#issuecomment-2631146036
'cuspatial/trajectory_clustering.ipynb',
]


Expand Down
6 changes: 3 additions & 3 deletions cuvs-bench/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ export DATA_FOLDER=path/to/store/results/and/data
docker run --gpus all --rm -it \
-v $DATA_FOLDER:/home/rapids/benchmarks \
-u $(id -u) \
rapidsai/cuvs-bench:25.02a-cuda12.5-py3.12 \
rapidsai/cuvs-bench:25.02a-cuda12.8-py3.12 \
"--dataset deep-image-96-angular" \
"--normalize" \
"--algorithms cuvs_cagra" \
Expand All @@ -47,7 +47,7 @@ Where:

- `DATA_FOLDER=path/to/store/results/and/data`: Results and datasets will be written to this host folder.
- `-u $(id -u)`: This flag allows the container to use the host user for permissions
- `rapidsai/cuvs-bench:25.02a-cuda12.5-py3.12`: Image to use, either `cuvs-bench` or `cuvs-bench-datasets`, cuVS version, CUDA version, and Python version.
- `rapidsai/cuvs-bench:25.02a-cuda12.8-py3.12`: Image to use, either `cuvs-bench` or `cuvs-bench-datasets`, cuVS version, CUDA version, and Python version.
- "--dataset deep-image-96-angular": Dataset name(s). See https://docs.rapids.ai/api/cuvs/nightly/cuvs_bench for more details.
- "--normalize": Whether to normalize the dataset, leave string empty ("") to not normalize.
- "--algorithms cuvs_cagra": What algorithm(s) to use as a ; separated list, as well as any other argument to pass to `cuvs_bench.run`.
Expand All @@ -74,7 +74,7 @@ export DATA_FOLDER=path/to/store/results/and/data
docker run --gpus all --rm -it \
-v $DATA_FOLDER:/home/rapids/benchmarks \
-u $(id -u) \
rapidsai/cuvs-bench:25.02a-cuda12.5-py3.12 \
rapidsai/cuvs-bench:25.02a-cuda12.8-py3.12 \
--entrypoint /bin/bash
```

Expand Down
8 changes: 4 additions & 4 deletions dockerhub-readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ There are two types:

The tag naming scheme for RAPIDS images incorporates key platform details into the tag as shown below:
```
25.02-cuda12.5-py3.12
25.02-cuda12.8-py3.12
^ ^ ^
| | Python version
| |
Expand All @@ -48,7 +48,7 @@ The tag naming scheme for RAPIDS images incorporates key platform details into t
RAPIDS version
```

**Note: Nightly builds of the images have the RAPIDS version appended with an `a` (ie `25.02a-cuda12.5-py3.12`)**
**Note: Nightly builds of the images have the RAPIDS version appended with an `a` (ie `25.02a-cuda12.8-py3.12`)**

## Usage

Expand Down Expand Up @@ -81,7 +81,7 @@ $ docker run \
-e EXTRA_CONDA_PACKAGES="jq" \
-e EXTRA_PIP_PACKAGES="beautifulsoup4" \
-p 8888:8888 \
rapidsai/notebooks:25.02-cuda12.5-py3.12
rapidsai/notebooks:25.02-cuda12.8-py3.12
```

### Bind Mounts
Expand All @@ -106,7 +106,7 @@ $ docker run \
--gpus all \
--shm-size=1g --ulimit memlock=-1 --ulimit stack=67108864 \
-v $(pwd)/environment.yml:/home/rapids/environment.yml \
rapidsai/base:25.02-cuda12.5-py3.12
rapidsai/base:25.02-cuda12.8-py3.12
```

### Use JupyterLab to Explore the Notebooks
Expand Down
6 changes: 3 additions & 3 deletions matrix-test.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,11 @@
pull-request:
- { CUDA_VER: '11.8', ARCH: 'amd64', PYTHON_VER: '3.10', GPU: 'v100', DRIVER: 'earliest' }
- { CUDA_VER: '12.0', ARCH: 'arm64', PYTHON_VER: '3.11', GPU: 'a100', DRIVER: 'latest' }
- { CUDA_VER: '12.5', ARCH: 'amd64', PYTHON_VER: '3.12', GPU: 'v100', DRIVER: 'latest' }
- { CUDA_VER: '12.8', ARCH: 'amd64', PYTHON_VER: '3.12', GPU: 'v100', DRIVER: 'latest' }
branch:
- { CUDA_VER: '11.8', ARCH: 'amd64', PYTHON_VER: '3.10', GPU: 'v100', DRIVER: 'earliest' }
- { CUDA_VER: '11.8', ARCH: 'amd64', PYTHON_VER: '3.10', GPU: 'v100', DRIVER: 'latest' }
- { CUDA_VER: '12.0', ARCH: 'amd64', PYTHON_VER: '3.11', GPU: 'v100', DRIVER: 'latest' }
- { CUDA_VER: '12.0', ARCH: 'arm64', PYTHON_VER: '3.11', GPU: 'a100', DRIVER: 'latest' }
- { CUDA_VER: '12.5', ARCH: 'amd64', PYTHON_VER: '3.12', GPU: 'v100', DRIVER: 'latest' }
- { CUDA_VER: '12.5', ARCH: 'arm64', PYTHON_VER: '3.12', GPU: 'a100', DRIVER: 'latest' }
- { CUDA_VER: '12.8', ARCH: 'amd64', PYTHON_VER: '3.12', GPU: 'v100', DRIVER: 'latest' }
- { CUDA_VER: '12.8', ARCH: 'arm64', PYTHON_VER: '3.12', GPU: 'a100', DRIVER: 'latest' }
4 changes: 2 additions & 2 deletions matrix.yaml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
CUDA_VER: # Should be `<major>.<minor>.<patch>` (e.g. `12.5.1`)
CUDA_VER: # Should be `<major>.<minor>.<patch>` (e.g. `12.8.0`)
- "11.8.0"
- "12.0.1"
- "12.5.1"
- "12.8.0"
PYTHON_VER:
- "3.10"
- "3.11"
Expand Down

0 comments on commit 3f8cea7

Please sign in to comment.