Commit ca734793 authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

fixed torch version to 1.4 to prevent HPC issues

parent 17bf6050
...@@ -13,7 +13,7 @@ setup( ...@@ -13,7 +13,7 @@ setup(
'nibabel', 'nibabel',
'numpy', 'numpy',
'pandas', 'pandas',
'torch', 'torch==1.4',
'h5py', 'h5py',
'fslpy', 'fslpy',
'tensorboardX', 'tensorboardX',
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment