RuntimeError: No such operator torchvision::nms

Yonggie發表於2020-12-04

我今天遇到了這個問題。
網上大部分的部落格,都要直接讓torch和torchvision降級去下載,我來說點人話。抄襲那些真的有意思嗎?呵呵。

我的問題具體觸發的原因呢,是我本來用的是cpu的torch,前些陣子直接換成了gpu的torch,由於是直接upgrade的,儘管torch本身是更新成了torch1.7.0+cu110,但是torchvision仍然是cpu版本的。
於是在執行之前寫好的程式碼的時候,就出現了上面的錯誤。

所以,說點人話,那就是你的torch和torchvision的版本要對應。我並沒有用大部分部落格的方法,我是直接升級了torchvision的版本,升級到對應於我torch1.7.0+cu110的版本。
升級的注意點在於,你不能直接
pip insatll upgrade xxx,因為沒有資源。
我是從pytorch官網上下載的,也就是
在這裡插入圖片描述使用的這個,升級完成後,完美執行。

相關文章