diff --git a/requirements.txt b/requirements.txt index d57db02..2e93e36 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ numpy==1.22.0 pickleshare==0.7.5 PyYAML==5.4.1 h5py==3.2.1 -torch==1.13.1 +torch==2.2.0 torchvision==0.8.2 setuptools>=65.5.1 # not directly required, pinned by Snyk to avoid a vulnerability wheel>=0.38.0 # not directly required, pinned by Snyk to avoid a vulnerability