[torch-frontend] update torch to 2.4.1

bytedance · Oct 23, 2024 · 06131ce · 06131ce
1 parent 4c96ed9
commit 06131ce
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/frontends/torch-frontend/scripts/envsetup.sh b/frontends/torch-frontend/scripts/envsetup.sh
@@ -33,7 +33,6 @@ function prepare_for_build_with_prebuilt() {
   pushd ${PROJ_DIR}
   # install requirements
   python3 -m pip install -r build-requirements.txt
-  # python3 -m pip install --no-cache-dir torch==2.1.0+cu118 torchvision==0.16.0+cu118 -f https://download.pytorch.org/whl/torch_stable.html
 
   # initialize submodule
   git submodule update --init -f $TORCH_MLIR_ROOT
@@ -49,10 +48,11 @@ function prepare_for_build() {
   pushd ${PROJ_DIR}
   # install requirements
   python3 -m pip install -r build-requirements.txt
-  # python3 -m pip install --no-cache-dir torch==2.1.0+cu118 torchvision==0.16.0+cu118 -f https://download.pytorch.org/whl/torch_stable.html
 
   # initialize submodule
   git submodule update --init --recursive -f $TORCH_MLIR_ROOT
 
   apply_patches
 }
+
+# python3 -m pip install --no-cache-dir torch==2.1.0+cu118 torchvision==0.16.0+cu118 -f https://download.pytorch.org/whl/torch_stable.html
diff --git a/frontends/torch-frontend/torch-cpu-requirements.txt b/frontends/torch-frontend/torch-cpu-requirements.txt
@@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
 --pre
-torch==2.1.0+cpu
-torchvision==0.16.0+cpu
+torch==2.4.1+cpu
+torchvision==0.19.1+cpu
diff --git a/frontends/torch-frontend/torch-cuda-requirements.txt b/frontends/torch-frontend/torch-cuda-requirements.txt
@@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/cu118
 --pre
-torch==2.1.0+cu118
-torchvision==0.16.0+cu118
+torch==2.4.1+cu118
+torchvision==0.19.1+cu118