Skip to content

Commit

Permalink
update pdm gpu cicd setup to pdm venv on nvme drive
Browse files Browse the repository at this point in the history
  • Loading branch information
Leif Denby committed Aug 13, 2024
1 parent 05f1e9f commit 3afe0e4
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions .github/workflows/ci-pdm-install-and-test-gpu.yml
Original file line number Diff line number Diff line change
Expand Up @@ -24,17 +24,19 @@ jobs:
- name: Create venv
run: |
pdm config venv.in_project False
pdm config venv.location /opt/dlami/nvme/venv
pdm venv create --with-pip
pdm use --venv in-project
- name: Install torch (GPU CUDA 12.1)
run: |
python -m pip install torch --index-url https://download.pytorch.org/whl/cu121
pdm run python -m pip install torch --index-url https://download.pytorch.org/whl/cu121
- name: Print and check torch version
run: |
python -c "import torch; print(torch.__version__)"
python -c "import torch; assert not torch.__version__.endswith('+cpu')"
pdm run python -c "import torch; print(torch.__version__)"
pdm run python -c "import torch; assert not torch.__version__.endswith('+cpu')"
- name: Setup tmate session
uses: mxschmitt/action-tmate@v1
Expand Down

0 comments on commit 3afe0e4

Please sign in to comment.