Skip to content

Commit

Permalink
Run torchbench on CI #9
Browse files Browse the repository at this point in the history
  • Loading branch information
DenisVieriu97 committed Mar 6, 2023
1 parent 9f943bf commit aca1452
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions .github/workflows/run_torchbench.yml
Original file line number Diff line number Diff line change
Expand Up @@ -38,9 +38,9 @@ jobs:
run: |
# shellcheck disable=SC1090
set -ex
${CONDA_RUN} conda install -y numpy="${NUMPY_VERSION}" requests ninja pyyaml \
setuptools cmake= typing-extensions boto3 \
pillow pytest tabulate gitpython git-lfs tqdm psutil
# ${CONDA_RUN} conda install -y numpy="${NUMPY_VERSION}" requests ninja pyyaml \
# setuptools cmake= typing-extensions boto3 \
# pillow pytest tabulate gitpython git-lfs tqdm psutil
${CONDA_RUN} pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
- name: Setup TorchBench branch
Expand Down

0 comments on commit aca1452

Please sign in to comment.