YOLOを使用して画像認識をやろうとして必要な環境構築をしようとした時に発生したエラーについての対処方法になります。
エラー内容
「Could not run 'torchvision::nms' with arguments from the 'CUDA' backend.」
経緯
YOLOに必要なものを一通りインストールしようとして、PCにPython(Ver.3.12.3)、Pytorch（以下のように公式サイトの通りPyTorch, Torchvision, Trochaudioをインストール）を入れてから、YOLOを実行してみたところ、上記のようなエラーが発生しました。
（このPyTorchのVer.は別PCでの表示例）
原因
pip listでバージョンを確認したところ、以下のようにtorchvisionだけGPUVer.になっていませんでした。
torch 2.3.0+cu118
torchaudio 2.3.0+cu118
torchvision 0.18.0
対処方法
そのため、以下のようにGPUVer.のものにインストールしなおしました。
pip install torchvision==0.18.0+cu118 -f
https://download.pytorch.org/whl/torch_stable.html
pip listでバージョンを確認したところ、以下のように正常になっていることが確認できました。
torch 2.3.0+cu118
torchaudio 2.3.0+cu118
torchvision 0.18.0+cu118
この状態で再度YOLOを実行したところ正常動作確認できました。
以上が、YOLOv8の環境構築時「Could not run 'torchvision::nms' with arguments from the 'CUDA' backend.」というエラーが出た時の対処方法になります。
