diff --git a/requirements.txt b/requirements.txt index 373137b..7ef55e6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ wget -torch==1.5.1 +torch==1.13.1 torchnet==0.0.4 torchvision==0.6.1 nibabel==3.0.2 diff --git a/setup.py b/setup.py old mode 100755 new mode 100644 index b520c41..114c325 --- a/setup.py +++ b/setup.py @@ -26,7 +26,7 @@ url="https://github.com/koriavinash1/DeepBrainSeg", packages=setuptools.find_packages(), install_requires = [ - 'torch==1.5.1', + 'torch==1.13.1', 'torchvision==0.6.1', 'torchnet==0.0.4', 'nibabel==3.0.2',