diff --git a/packaging/pre_build_script.sh b/packaging/pre_build_script.sh index d975e4cbe3..911807f364 100755 --- a/packaging/pre_build_script.sh +++ b/packaging/pre_build_script.sh @@ -13,7 +13,9 @@ wget https://github.com/bazelbuild/bazelisk/releases/download/v1.17.0/bazelisk-l TORCH_TORCHVISION=$(grep "^torch" py/requirements.txt) INDEX_URL=https://download.pytorch.org/whl/${CHANNEL}/${CU_VERSION} + # Install all the dependencies required for Torch-TensorRT +pip uninstall torch torchvision pip install --force-reinstall --pre ${TORCH_TORCHVISION} --index-url ${INDEX_URL} pip install --pre -r tests/py/requirements.txt --use-deprecated legacy-resolver diff --git a/packaging/pre_build_script_windows.sh b/packaging/pre_build_script_windows.sh index a8e9dc0ff2..b37168d779 100644 --- a/packaging/pre_build_script_windows.sh +++ b/packaging/pre_build_script_windows.sh @@ -12,7 +12,9 @@ choco install bazelisk -y #unzip -o TensorRT.zip -d C:/ TORCH_TORCHVISION=$(grep "^torch" py/requirements.txt) INDEX_URL=https://download.pytorch.org/whl/${CHANNEL}/${CU_VERSION} + # Install all the dependencies required for Torch-TensorRT +pip uninstall torch torchvision pip install --force-reinstall --pre ${TORCH_TORCHVISION} --index-url ${INDEX_URL} pip install --pre -r tests/py/requirements.txt --use-deprecated legacy-resolver