Skip to content

Commit

Permalink
Update CI
Browse files Browse the repository at this point in the history
  • Loading branch information
RaulPPelaez committed Nov 17, 2023
1 parent e9e6a6d commit d7075e8
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions .github/workflows/ci-cuda12.yml
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,6 @@ jobs:
include:
# Latest supported versions (with CUDA)
- name: Linux (CUDA 12, Python 3.11, PyTorch 2.1)
enable_cuda: true
cuda: "12"
gcc: "11.*"
python: "3.11.*"
Expand All @@ -42,7 +41,6 @@ jobs:
miniforge-variant: Mambaforge

- name: Prepare dependencies (with CUDA)
if: ${{ matrix.enable_cuda }}
run: |
sed -i -e "/cudatoolkit/c\ - cuda ${{ matrix.cuda }}" \
-e "/gxx_linux-64/c\ - gxx ${{ matrix.gcc }}" \
Expand All @@ -56,7 +54,10 @@ jobs:
run: cat environment.yml

- name: Install dependencies
run: CONDA_OVERRIDE_CUDA=${{ matrix.nvcc }} mamba env create -n nnpops -f environment.yml
run: mamba env create -n nnpops -f environment.yml
env:
# Needed to install pytorch-gpu on a machine without a GPU
CONDA_OVERRIDE_CUDA: ${{ matrix.cuda }}

- name: List conda environment
run: |
Expand Down

0 comments on commit d7075e8

Please sign in to comment.