极链ai云使用RTX3090报错:RuntimeError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA
出现RuntimeError: Could not run ‘torchvision::nms’ with arguments from the 'CUDA的错误。经过查询,是因为torch、torchvision和CUDA版本的问题,后来使用该文章中添加链接描述的配置解决了问题。(自己选用的是CUDA11.0,刚好是一致的!!)...
原创
2022-06-30 15:18:29 ·
1377 阅读 ·
0 评论