File tree Expand file tree Collapse file tree 2 files changed +1
-7
lines changed Expand file tree Collapse file tree 2 files changed +1
-7
lines changed Original file line number Diff line number Diff line change @@ -3,7 +3,6 @@ set -eou pipefail
3
3
# Source conda so it's available to the script environment
4
4
source ${BUILD_ENV_FILE}
5
5
export EXTRA_INDEX_URL=" https://download.pytorch.org/whl/test/${CU_VERSION} "
6
- export PLATFORM=$( ${CONDA_RUN} python -c " import sys; print(sys.platform)" )
7
6
# Install all the dependencies required for Torch-TensorRT
8
7
${CONDA_RUN} pip install --pre -r ${PWD} /tests/py/requirements.txt --use-deprecated=legacy-resolver --extra-index-url=${EXTRA_INDEX_URL}
9
8
Original file line number Diff line number Diff line change @@ -3,15 +3,10 @@ set -eou pipefail
3
3
# Source conda so it's available to the script environment
4
4
source ${BUILD_ENV_FILE}
5
5
export EXTRA_INDEX_URL=" https://download.pytorch.org/whl/test/${CU_VERSION} "
6
- export PLATFORM=$( ${CONDA_RUN} python -c " import sys; print(sys.platform)" )
7
6
# Install all the dependencies required for Torch-TensorRT
8
7
${CONDA_RUN} pip install --pre -r ${PWD} /tests/py/requirements.txt --use-deprecated=legacy-resolver --extra-index-url=${EXTRA_INDEX_URL}
9
8
10
9
# 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
16
11
17
12
echo -e " Running test script" ;
You can’t perform that action at this time.
0 commit comments