diff --git a/docker/install/ubuntu_install_onnx.sh b/docker/install/ubuntu_install_onnx.sh
index 2bb50c619815..6cea0075c102 100755
--- a/docker/install/ubuntu_install_onnx.sh
+++ b/docker/install/ubuntu_install_onnx.sh
@@ -36,6 +36,6 @@ pip3 install \
 pip3 install future
 
 pip3 install \
-    torch==2.0.0 \
-    torchvision==0.15.1 \
+    torch==2.4.1 \
+    torchvision==0.19.1 \
     --extra-index-url https://download.pytorch.org/whl/cpu