UWSOD icon indicating copy to clipboard operation
UWSOD copied to clipboard

Can it run on pytorch1.7 cuda11.0

Open Horatio-Ng opened this issue 3 years ago • 3 comments

Hi~ I only have an NVIDIA 3090, so can the code run on pytorch1.7 with cuda11.0?

Horatio-Ng avatar Feb 02 '21 08:02 Horatio-Ng

Although I do not test it on pytorch1.7 with cuda11.0, I think it should work. If you have any problem please feel free to open issues.

shenyunhang avatar Feb 02 '21 13:02 shenyunhang

It seems that UWSOD/projects/WSL/setup.py needs to be modified to make it suitable for pytorch1.7. However, I still don't know how to modify it~

Horatio-Ng avatar Feb 02 '21 17:02 Horatio-Ng

I just have installed it with PyTorch 1.7. Can you post the installing log?

shenyunhang avatar Feb 03 '21 03:02 shenyunhang