From 1b57bdb1ecc9728bd1a7590e41eb942998b8c212 Mon Sep 17 00:00:00 2001 From: Yousef Moazzam Date: Tue, 17 Dec 2024 10:09:09 +0000 Subject: [PATCH] Update CUDA major version in IRIS CI container to 12 --- .github/workflows/run_tests_iris.yml | 2 +- pyproject.toml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/run_tests_iris.yml b/.github/workflows/run_tests_iris.yml index f330e15..c47030f 100644 --- a/.github/workflows/run_tests_iris.yml +++ b/.github/workflows/run_tests_iris.yml @@ -12,7 +12,7 @@ jobs: iris-gpu: runs-on: iris-gpu container: - image: nvidia/cuda:11.6.2-devel-ubi8 + image: nvidia/cuda:12.6.3-devel-ubi8 env: NVIDIA_VISIBLE_DEVICES: ${{ env.NVIDIA_VISIBLE_DEVICES }} diff --git a/pyproject.toml b/pyproject.toml index 5a23ae6..0f54607 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -50,7 +50,7 @@ dependencies = [ "numpy", "pyyaml", "tomopy", - "cupy-cuda11x==12.3.0", + "cupy-cuda12x==12.3.0", "nvtx", "toml", "imageio",