Skip to content

Commit

Permalink
Change torch and fbgemm-gpu version
Browse files Browse the repository at this point in the history
Summary: Version mismatch between fbgemm-gpu and torch can cause errors. Debug to see if the problem persists if the compatible version (i.e., nightly) is used.

Differential Revision: D48460839

fbshipit-source-id: 58d7e67723d1fcb0dc173fd1b386c28e67346f83
  • Loading branch information
spcyppt authored and facebook-github-bot committed Aug 18, 2023
1 parent 77ec968 commit dc5a7b7
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions .github/workflows/run_tests.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -73,12 +73,12 @@ jobs:
run: |
set -eux
conda activate test
pip install torch --extra-index-url https://download.pytorch.org/whl/cu116
pip install torch --extra-index-url https://download.pytorch.org/whl/cu117
pip install torchrec-nightly
# Use stable fbgemm-gpu
pip uninstall -y fbgemm-gpu-nightly
pip install fbgemm-gpu
# pip uninstall -y fbgemm-gpu-nightly
pip install fbgemm-gpu-nightly
pip install -r requirements.txt
pip install -r dev-requirements.txt
Expand Down

0 comments on commit dc5a7b7

Please sign in to comment.