test

pytorch · Aug 15, 2024 · cc07f3d · cc07f3d
1 parent 68645c3
commit cc07f3d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git 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
@@ -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