From e0c2162591eac7ea03f657ff9ed6c5048771dcb8 Mon Sep 17 00:00:00 2001 From: Michael McKinsey Date: Fri, 31 Jul 2026 11:54:55 -0700 Subject: [PATCH] Update version --- pyproject.toml | 2 +- scripts/install-matrix.sh | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index c0e6c3e..3a4c877 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -74,7 +74,7 @@ test = [ "psutil", ] cuda = [ - "torch==2.10.0+cu129", + "torch==2.12.0+cu132", "mpi4py==4.1.1", ] rocm = [ diff --git a/scripts/install-matrix.sh b/scripts/install-matrix.sh index 15c4e6d..eb6b782 100644 --- a/scripts/install-matrix.sh +++ b/scripts/install-matrix.sh @@ -1,4 +1,4 @@ -ml load python/3.11.5 && python3 -m venv .venvs/scaffoldvenv-matrix && source .venvs/scaffoldvenv-matrix/bin/activate && pip install --upgrade pip -ml cuda/12.9.1 gcc/13.3.1 mvapich2/2.3.7 +ml load python/3.13.2 && python3 -m venv .venvs/scaffoldvenv-matrix && source .venvs/scaffoldvenv-matrix/bin/activate && pip install --upgrade pip +ml cuda/13.1.1 gcc/13.3.1 mvapich2/2.3.7 export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH -pip install --no-binary=mpi4py -e .[cuda] --prefix=.venvs/scaffoldvenv-matrix --extra-index-url https://download.pytorch.org/whl/cu129 2>&1 | tee install.log +pip install --no-binary=mpi4py -e .[cuda] --prefix=.venvs/scaffoldvenv-matrix --extra-index-url https://download.pytorch.org/whl/cu132 2>&1 | tee install.log