diff --git a/requirements.txt b/requirements.txt index 0e24fbf..0f04b81 100644 --- a/requirements.txt +++ b/requirements.txt @@ -23,7 +23,7 @@ SimpleITK tensorflow>=1.14,<2 # The book was written targeting PyTorch 1.4, but everything in it should work fine with more up-to-date versions. -torch==1.4 +torch==2.2.0 # future is here due to a missing dep in caffe2 # future