diff --git a/pytorch-inference-docker-lambda/requirements.txt b/pytorch-inference-docker-lambda/requirements.txt index 986f2c2..6f75533 100644 --- a/pytorch-inference-docker-lambda/requirements.txt +++ b/pytorch-inference-docker-lambda/requirements.txt @@ -1,4 +1,4 @@ --find-links https://download.pytorch.org/whl/torch_stable.html -torch==1.7.1+cpu +torch==1.13.1 torchvision==0.8.2+cpu \ No newline at end of file