Skip to content

Commit

Permalink
switch from 118 to 121
Browse files Browse the repository at this point in the history
  • Loading branch information
henrylhtsang committed Oct 4, 2023
1 parent d8be0b2 commit a4231c9
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions .github/workflows/release_build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -25,19 +25,19 @@ jobs:
- os: linux.2xlarge
python-version: 3.8
python-tag: "py38"
cuda-tag: "cu118"
cuda-tag: "cu121"
- os: linux.2xlarge
python-version: 3.9
python-tag: "py39"
cuda-tag: "cu118"
cuda-tag: "cu121"
- os: linux.2xlarge
python-version: '3.10'
python-tag: "py310"
cuda-tag: "cu118"
cuda-tag: "cu121"
- os: linux.2xlarge
python-version: '3.11'
python-tag: "py311"
cuda-tag: "cu118"
cuda-tag: "cu121"
steps:
# Checkout the repository to the GitHub Actions runner
- name: Check ldd --version
Expand Down Expand Up @@ -73,11 +73,11 @@ jobs:
- name: Install PyTorch and CUDA
shell: bash
run: |
conda run -n build_binary pip install torch --index-url https://download.pytorch.org/whl/test/cu118
conda run -n build_binary pip install torch --index-url https://download.pytorch.org/whl/test/cu121
- name: Install fbgemm
shell: bash
run: |
conda run -n build_binary pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/test/cu118
conda run -n build_binary pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/test/cu121
- name: Install Dependencies
shell: bash
run: |
Expand Down Expand Up @@ -111,7 +111,7 @@ jobs:
matrix:
os: [linux.4xlarge.nvidia.gpu]
python-version: [3.8, 3.9, "3.10", "3.11"]
cuda-tag: ["cu118"]
cuda-tag: ["cu121"]
needs: build_on_cpu
# the glibc version should match the version of the one we used to build the binary
# for this case, it's 2.26
Expand Down Expand Up @@ -176,12 +176,12 @@ jobs:
- name: Install PyTorch and CUDA
shell: bash
run: |
conda run -n build_binary pip install torch --index-url https://download.pytorch.org/whl/test/cu118
conda run -n build_binary pip install torch --index-url https://download.pytorch.org/whl/test/cu121
# download wheel from GHA
- name: Install fbgemm
shell: bash
run: |
conda run -n build_binary pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/test/cu118
conda run -n build_binary pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/test/cu121
- name: Install torchmetrics
shell: bash
run: |
Expand Down

0 comments on commit a4231c9

Please sign in to comment.