From ebbf294aa0bf75c27b64e4d83bb8bc32fced2884 Mon Sep 17 00:00:00 2001 From: Masahiro Hiramori Date: Thu, 5 Sep 2024 17:36:23 +0900 Subject: [PATCH] upgrade pytorch to 2.4.1 --- docker/install/ubuntu_install_onnx.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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