diff --git a/frontends/torch-frontend/scripts/envsetup.sh b/frontends/torch-frontend/scripts/envsetup.sh index d9dc80c28..982e37441 100755 --- 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 index 15cd43740..57ea64eae 100644 --- 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 index 7be5d08b3..5cae3892e 100644 --- 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