diff --git a/.github/workflows/conda_gpu_nigthly.yaml b/.github/workflows/conda_gpu_nigthly.yaml index 0e30680..2f0e8d5 100644 --- a/.github/workflows/conda_gpu_nigthly.yaml +++ b/.github/workflows/conda_gpu_nigthly.yaml @@ -17,16 +17,16 @@ jobs: pkg: ['tlcpack', 'tlcpack-nightly'] config: - cuda: '10.2' - image: 'tlcpack/package-cu102:7bfdf7e' + image: 'tlcpack/package-cu102:2fddbe0' # Conda forge is not shipping a cudnn package with support for CUDA 11.0 # - cuda: '11.0' - # image: 'tlcpack/package-cu110:7bfdf7e' + # image: 'tlcpack/package-cu110:2fddbe0' - cuda: '11.1' - image: 'tlcpack/package-cu111:7bfdf7e' + image: 'tlcpack/package-cu111:2fddbe0' - cuda: '11.3' - image: 'tlcpack/package-cu113:7bfdf7e' + image: 'tlcpack/package-cu113:2fddbe0' - cuda: '11.6' - image: 'tlcpack/package-cu116:7bfdf7e' + image: 'tlcpack/package-cu116:2fddbe0' runs-on: ubuntu-latest diff --git a/.github/workflows/wheel_manylinux_nightly.yaml b/.github/workflows/wheel_manylinux_nightly.yaml index 0b054e8..64fce0c 100644 --- a/.github/workflows/wheel_manylinux_nightly.yaml +++ b/.github/workflows/wheel_manylinux_nightly.yaml @@ -19,19 +19,19 @@ jobs: # matrix of build configs config: - cuda: 'none' - image: 'tlcpack/package-cpu:7bfdf7e' + image: 'tlcpack/package-cpu:2fddbe0' - cuda: '10.2' - image: 'tlcpack/package-cu102:7bfdf7e' + image: 'tlcpack/package-cu102:2fddbe0' # CUDA 11.0 is not supported for Conda build, # Let's drop support here as well. # - cuda: '11.0' - # image: 'tlcpack/package-cu110:7bfdf7e' + # image: 'tlcpack/package-cu110:2fddbe0' - cuda: '11.1' - image: 'tlcpack/package-cu111:7bfdf7e' + image: 'tlcpack/package-cu111:2fddbe0' - cuda: '11.3' - image: 'tlcpack/package-cu113:7bfdf7e' + image: 'tlcpack/package-cu113:2fddbe0' - cuda: '11.6' - image: 'tlcpack/package-cu116:7bfdf7e' + image: 'tlcpack/package-cu116:2fddbe0' runs-on: ubuntu-latest diff --git a/.github/workflows/wheel_manylinux_pypi.yaml b/.github/workflows/wheel_manylinux_pypi.yaml index 761a978..a918203 100644 --- a/.github/workflows/wheel_manylinux_pypi.yaml +++ b/.github/workflows/wheel_manylinux_pypi.yaml @@ -20,7 +20,7 @@ jobs: # matrix of build configs config: - cuda: 'none' - image: 'tlcpack/package-cpu:7bfdf7e' + image: 'tlcpack/package-cpu:2fddbe0' package_name: 'apache-tvm' runs-on: ubuntu-latest