attp 10

Author: calderjo

Jenkinsfile

@@ -64,7 +64,7 @@ pipeline {
         }
         stage('jaxlib') {
           options {
-            timeout(time: 300, unit: 'MINUTES')
+            timeout(time: 500, unit: 'MINUTES')
           }
           steps {
             sh '''#!/bin/bash

packages/torch.Dockerfile

@@ -20,7 +20,7 @@ RUN conda install -c conda-forge mamba
 
 # Build instructions: https://github.com/pytorch/pytorch#from-source
 RUN mamba install astunparse numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses
-RUN mamba install -c pytorch magma-cuda${CUDA_MAJOR_VERSION}${CUDA_MINOR_VERSION}
+RUN mamba install -c pytorch magma-cuda122
 
 # By default, it uses the version from version.txt which includes the `a0` (alpha zero) suffix and part of the git hash.
 # This causes dependency conflicts like these: https://paste.googleplex.com/4786486378496000