UaNet icon indicating copy to clipboard operation
UaNet copied to clipboard

Published 20 hours ago • verified publisher icon uci-cbcl

Compatibility with pytorch 1.5 & CUDA 10.2

Open jmarsil opened this issue 4 years ago • 2 comments

Just wondering if work has been done to update the scripts to support the newest versions of PyTorch and CUDA? I got the following error when trying to install the custom module for bounding box NMS and overlap calculation. I assume this is a required package for bounding box calculation applied in brain_reader.py?

Cuda Version: 10.2 Pytorch Version: 1.5

Screen Shot 2020-07-11 at 9 04 34 PM

Any help would be greatly appreciated!

jmarsil avatar Jul 12 '20 01:07 jmarsil

Hi, do you solve this problem?

Suiiyu avatar Mar 03 '22 12:03 Suiiyu

AT_CHECK is deprecated in torch 1.5. You can replace AT_CHECK in nms.cpp and overlap.cpp with TORCH_CHECK

Suiiyu avatar Mar 04 '22 06:03 Suiiyu