search

MhLiao / MaskTextSpotterV3

The code of "Mask TextSpotter v3: Segmentation Proposal Network for Robust Scene Text Spotting"
Other
622 stars 122 forks source link

Trying to build using Cuda 11.0 #48

Open Foglia-m opened 3 years ago

Foglia-m commented 3 years ago

I used to run mask text spotter on a 1080ti and had great results but I recently switched to rtx 3090 (ampere architecture), therefore I had to build using pytorch with cuda 11.0 which doesn't seem to support apex (It seems that it's supported directly within Pytorch now: https://github.com/NVIDIA/apex/issues/988#issuecomment-726343453) Has anyone experienced theses issues and solved it?
Perhaps there is a workaround to change some code in the masktextspotter build to use pytorch amp (torch.cuda.amp). instead of apex's. I'd like to discuss this subject with anyone interested.

qiuchun commented 2 years ago

hi i also have this problem , whether you have sloved this problem？ can you help me ? think you !