From a2cdf3a17e4722ceacab36c55e0c5cb7958f025c Mon Sep 17 00:00:00 2001 From: Nathaniel McVicar Date: Mon, 16 Oct 2023 18:05:15 +0000 Subject: [PATCH] Fix python install location --- .github/workflows/torchbench.yml | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/.github/workflows/torchbench.yml b/.github/workflows/torchbench.yml index 3ee0ab6fc007..6e474eb9eaf6 100644 --- a/.github/workflows/torchbench.yml +++ b/.github/workflows/torchbench.yml @@ -29,14 +29,18 @@ jobs: # Force user install, TorchBench expects a conda environment - name: Install TorchBench and models run: | - echo -e "[install]\nprefix=${HOME}/.local\n" > ${HOME}/.pydistutils.cfg + cd python + ls python3 -m pip install --upgrade pip - python3 -m pip install --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 + python3 -m pip install --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 df -h du --max-depth=1 -h - cd torchbench + cd ../torchbench + echo -e "[install]\nprefix=${HOME}/.local\n" > ${HOME}/.pydistutils.cfg python3 install.py detectron2_fasterrcnn_r_101_c4 - + df -h + du --max-depth=1 -h + - name: Install the latest Triton wheel run: | python3 -m pip install --upgrade triton-nightly --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/Triton-Nightly/pypi/simple/