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(
'nibabel',
'numpy',
'pandas',
'torch',
'torch==1.4',
'h5py',
'fslpy',
'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