test

pytorch · Aug 15, 2024 · 90dc93f · 90dc93f
1 parent cc07f3d
commit 90dc93f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/packaging/pre_build_script.sh b/packaging/pre_build_script.sh
@@ -15,7 +15,7 @@ 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 uninstall -y 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
@@ -14,7 +14,7 @@ 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 uninstall -y 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