Skip to content

Commit

Permalink
Fix github action CI for GPU (#625)
Browse files Browse the repository at this point in the history
Summary: Pull Request resolved: #625

Differential Revision: D53292685
  • Loading branch information
HuanyuZhang authored and facebook-github-bot committed Feb 14, 2024
1 parent 1ba4113 commit 7d8130c
Showing 1 changed file with 11 additions and 10 deletions.
21 changes: 11 additions & 10 deletions .github/workflows/ci_gpu.yml
Original file line number Diff line number Diff line change
Expand Up @@ -8,16 +8,17 @@ on:
branches:
- main

jobs:
unittest_multi_gpu:
runs-on: linux.4xlarge.nvidia.gpu
runs-on: 4-core-ubuntu-gpu-t4
steps:
- name: Checkout
uses: actions/checkout@v2

- name: Set up Python
uses: actions/setup-python@v2
with:
python-version: 3.9
python-version: '3.9'

- name: Install dependencies
run: |
Expand All @@ -31,21 +32,21 @@ on:
integrationtest_py39_torch_release_cuda:
runs-on: ubuntu-latest
container:
# https://hub.docker.com/r/nvidia/cuda
image: nvidia/cuda:12.3.1-base-ubuntu22.04
options: --gpus all
env:
TZ: 'UTC'
runs-on: linux.4xlarge.nvidia.gpu
# container:
# # https://hub.docker.com/r/nvidia/cuda
# image: nvidia/cuda:12.3.1-base-ubuntu22.04
# options: --gpus all
# env:
# TZ: 'UTC'
steps:
- name: Checkout
uses: actions/checkout@v2

- name: Set up Python
uses: actions/setup-python@v2
with:
python-version: 3.9
python-version: '3.10.13'

- name: Install dependencies
run: |
Expand Down

0 comments on commit 7d8130c

Please sign in to comment.