最近YOLOなどを使って画像認識をやろうとしているのですがその時に発生したエラーとその対応方法を備忘録として載せておきます。
YOLOのモデルを使ってセグメンテーションを実行しようとした時の初期のエラーとなりますが以下のようなエラーが発生しました。
AssertionError: Torch not compiled with CUDA enabled
使っていたのはWindowsマシンですが、WindowsがGPU（CUDA）を認識しない時にこのようなエラーが発生します。
今回の場合は、PyTorchの中にCPUのみのバージョンが存在しているためです。
conda環境ではない場合の対応方法になりますが以下のようにtorchをインストールし直すということになります。
torchをアンインストールした後、
pip uninstall torch
キャッシュをクリアして、
pip cache purge
torchを再インストールします。
pip install torch -f https://download.pytorch.org/whl/torch_stable.html
これが今回発生したエラーを解決するための方法の1つになります。
ただし、自分達の場合はGPU対応ではないPCでやろうとしていたというそれ以前の問題であったため、以下の中の「cuda」を「cpu」に変更することで解決しました。
from ultralytics import YOLO
# モデルを選択
model = YOLO("yolov8n.pt")
model.to("cuda")
results = model("./datasets/images/test.mp4", save=True)
for i in enumerate(results):
print(i)
以上が、WindowsでGPU（CUDA）を認識しない時に 「AssertionError: Torch not compiled with CUDA enabled 」というエラーが発生した時の対応方法になります。
【Python】Python仮想環境を別PCに移設する方法 - Django Girls and Boys 備忘録
【Python】Pythonで仮想環境を構築する手順 - Django Girls and Boys 備忘録
【Python 仮想環境構築】OneDriveに仮想環境を構築する方法 - Django Girls and Boys 備忘録
【Python】subprocess.Popen()を使ったプロセスの並列起動について - Django Girls and Boys 備忘録
【Python】exeファイルを実行する方法：subprocess.run()を使った簡単な手順 - Django Girls and Boys 備忘録
プログラミング初心者に最適な一冊：『いちばんやさしいGit&GitHubの教本 第2版』 - Django Girls and Boys 備忘録
【Python】フォルダ内にあるファイル名を抽出する方法 - Django Girls and Boys 備忘録
【Python】キーの同時押し、連続操作などの実施方法 - Django Girls and Boys 備忘録
【Python】PythonでPdfWriterを使ってPDFファイルを分割する方法 - Django Girls and Boys 備忘録
