From ae15bf09d56f5b0c19ae1af3fddf15f8ea7acfbd Mon Sep 17 00:00:00 2001 From: Joel Polizzi Date: Fri, 10 Nov 2023 11:28:50 -0800 Subject: [PATCH] Revert the change in onnx-gpu. In 1.9.1 there is no cu111 package. Signed-off-by: Joel Polizzi --- Jenkins/Dockerfile.onnx-gpu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Jenkins/Dockerfile.onnx-gpu b/Jenkins/Dockerfile.onnx-gpu index ad6f7e15af5..06d481ad312 100644 --- a/Jenkins/Dockerfile.onnx-gpu +++ b/Jenkins/Dockerfile.onnx-gpu @@ -191,7 +191,7 @@ RUN python3 -m pip --no-cache-dir install \ tensorboardX==2.4 \ timm==0.4.12 \ torch==1.9.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html \ - torchaudio==0.9.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html \ + torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html \ torchtext==0.10.1 \ torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html \ tqdm \