Skip to content

Commit 05bf78b

Browse files
fix ci broken issue (#3931)
1 parent a80572d commit 05bf78b

File tree

5 files changed

+6
-3
lines changed

5 files changed

+6
-3
lines changed

.github/scripts/install-torch-tensorrt.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@ pip install --pre -r ${PWD}/tests/py/requirements.txt
1818
# eg. timm will install the latest torchvision, however we want to use the torchvision from nightly
1919
# reinstall torch torchvisionto make sure we have the correct version
2020
pip uninstall -y torch torchvision
21-
pip install --force-reinstall --pre ${TORCHVISION} --index-url ${INDEX_URL}
22-
pip install --force-reinstall --pre ${TORCH} --index-url ${INDEX_URL}
21+
pip install --force-reinstall --pre ${TORCHVISION} --index-url ${INDEX_URL} --extra-index-url https://pypi.org/simple
22+
pip install --force-reinstall --pre ${TORCH} --index-url ${INDEX_URL} --extra-index-url https://pypi.org/simple
2323

2424
if [[ ${USE_TRT_RTX} == true ]]; then
2525
source .github/scripts/install-tensorrt-rtx.sh

.github/workflows/build-test-linux-aarch64.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ jobs:
7777
smoke-test-script: ${{ matrix.smoke-test-script }}
7878
trigger-event: ${{ github.event_name }}
7979
architecture: "aarch64"
80+
pip-install-torch-extra-args: "--extra-index-url https://pypi.org/simple"
8081

8182
concurrency:
8283
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}-${{ inputs.job-name }}

.github/workflows/build-test-linux-x86_64.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ jobs:
7777
trigger-event: ${{ github.event_name }}
7878
architecture: "x86_64"
7979
use-rtx: false
80+
pip-install-torch-extra-args: "--extra-index-url https://pypi.org/simple"
8081

8182
L0-dynamo-converter-tests:
8283
name: L0 dynamo converter tests

.github/workflows/build-test-linux-x86_64_rtx.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,7 @@ jobs:
7676
trigger-event: ${{ github.event_name }}
7777
architecture: "x86_64"
7878
use-rtx: true
79+
pip-install-torch-extra-args: "--extra-index-url https://pypi.org/simple"
7980

8081
L0-dynamo-converter-tests:
8182
name: L0 dynamo converter tests

packaging/pre_build_script.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ else
5353
INDEX_URL=https://download.pytorch.org/whl/${CHANNEL}/${CU_VERSION}
5454

5555
# Install all the dependencies required for Torch-TensorRT
56-
pip install --force-reinstall --pre ${TORCH} --index-url ${INDEX_URL}
56+
pip install --force-reinstall --pre ${TORCH} --index-url ${INDEX_URL} --extra-index-url https://pypi.org/simple
5757
fi
5858

5959
export TORCH_BUILD_NUMBER=$(python -c "import torch, urllib.parse as ul; print(ul.quote_plus(torch.__version__))")

0 commit comments

Comments
 (0)