Skip to content

Commit

Permalink
fix(ci): Prevent CUDA installation in lint action
Browse files Browse the repository at this point in the history
  • Loading branch information
cpvannier committed Feb 28, 2024
1 parent e618cdc commit bedd35b
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion .github/workflows/lint.yml
Original file line number Diff line number Diff line change
Expand Up @@ -40,11 +40,13 @@ jobs:
with:
python-version: "3.10"

# Install PyTorch and TensorFlow CPU versions manually to prevent installing CUDA
- name: Install dependencies
run: |
python -m pip install --upgrade pip
python -m pip install pylint
python -m pip install .
python -m pip install torch~=2.2.0 torchaudio~=2.2.0 torchvision~=0.17.0 --index-url https://download.pytorch.org/whl/cpu
python -m pip install tensorflow-cpu~=2.15.0
- name: Lint backend code with Pylint
run: |
Expand Down

0 comments on commit bedd35b

Please sign in to comment.