diff --git a/examples/ML+DL-Examples/Spark-DL/dl_inference/torch_requirements.txt b/examples/ML+DL-Examples/Spark-DL/dl_inference/torch_requirements.txt index bf836c02..f8980253 100644 --- a/examples/ML+DL-Examples/Spark-DL/dl_inference/torch_requirements.txt +++ b/examples/ML+DL-Examples/Spark-DL/dl_inference/torch_requirements.txt @@ -13,7 +13,7 @@ # limitations under the License. -r requirements.txt -torch==2.8.0 +torch==2.12.1 torchvision==0.23.0 torch-tensorrt==2.8.0 tensorrt --extra-index-url https://download.pytorch.org/whl/cu121