BUG: UserWarning: No NMS is available. Please upgrade torchvision to 0.3.0+

[xmdszzz]

The TorchVision's version check uses dictionary order. When the version number is '0.17.0+cpu', an incorrect judgment occurs, causing the code to enter the else branch, throwing an error and interrupting the program abnormally. incorrect code in ssd/utils/nms.py:

if torchvision.__version__ >= '0.3.0':
    _nms = torchvision.ops.nms
else:
    warnings.warn('No NMS is available. Please upgrade torchvision to 0.3.0+')
    sys.exit(-1)

my suggestion is import package : packaging, and than:

from packaging import version
if version.parse(torchvision.__version__) >= version.parse('0.3.0'):
    _nms = torchvision.ops.nms
else:
    warnings.warn('No NMS is available. Please upgrade torchvision to 0.3.0+')
    sys.exit(-1)

[ax1an9]

I think this works.