ci: update torch for py310 compat

flatironinstitute · Jul 23, 2024 · 9179701 · 9179701
1 parent 8ef919c
commit 9179701
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Jenkinsfile b/Jenkinsfile
@@ -58,7 +58,7 @@ pipeline {
     sh '''#!/bin/bash -ex
       source $HOME/bin/activate
       python3 -m pip install --no-cache-dir --upgrade pycuda cupy-cuda112 numba
-      python3 -m pip install --no-cache-dir torch==1.10.2+cu111 -f https://download.pytorch.org/whl/torch_stable.html
+      python3 -m pip install --no-cache-dir torch==1.12.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html
       python3 -m pip install --no-cache-dir pytest
       python -c "from numba import cuda; cuda.cudadrv.libs.test()"
       python3 -m pytest --framework=pycuda python/cufinufft