Revert the change in onnx-gpu. In 1.9.1 there is no cu111 package.

Signed-off-by: Joel Polizzi <quic_jpolizzi@quicinc.com>
quic · Nov 15, 2023 · ae15bf0 · ae15bf0
1 parent 6333d77
commit ae15bf0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 \