Skip to content

Commit a983064

Browse files
committed
chore: updates
1 parent c71c017 commit a983064

File tree

2 files changed

+1
-7
lines changed

2 files changed

+1
-7
lines changed

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

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@ set -eou pipefail
33
# Source conda so it's available to the script environment
44
source ${BUILD_ENV_FILE}
55
export EXTRA_INDEX_URL="https://download.pytorch.org/whl/test/${CU_VERSION}"
6-
export PLATFORM=$(${CONDA_RUN} python -c "import sys; print(sys.platform)")
76
# Install all the dependencies required for Torch-TensorRT
87
${CONDA_RUN} pip install --pre -r ${PWD}/tests/py/requirements.txt --use-deprecated=legacy-resolver --extra-index-url=${EXTRA_INDEX_URL}
98

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

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,15 +3,10 @@ set -eou pipefail
33
# Source conda so it's available to the script environment
44
source ${BUILD_ENV_FILE}
55
export EXTRA_INDEX_URL="https://download.pytorch.org/whl/test/${CU_VERSION}"
6-
export PLATFORM=$(${CONDA_RUN} python -c "import sys; print(sys.platform)")
76
# Install all the dependencies required for Torch-TensorRT
87
${CONDA_RUN} pip install --pre -r ${PWD}/tests/py/requirements.txt --use-deprecated=legacy-resolver --extra-index-url=${EXTRA_INDEX_URL}
98

109
# Install Torch-TensorRT via pre-built wheels. On windows, the location of wheels is not fixed.
11-
if [[ "$PLATFORM" == "win32" ]]; then
12-
${CONDA_RUN} pip install ${RUNNER_ARTIFACT_DIR}/torch_tensorrt*.whl
13-
else
14-
${CONDA_RUN} pip install /opt/torch-tensorrt-builds/torch_tensorrt*.whl
15-
fi
10+
${CONDA_RUN} pip install /opt/torch-tensorrt-builds/torch_tensorrt*.whl
1611

1712
echo -e "Running test script";

0 commit comments

Comments
 (0)