環境
OS : windows10
python : 3.11.15
pytorch : 2.4.0
仮想環境 : venv
問題
yoloを使う際にpytorchが必要になりpip install torch torchvision torchaudio
でtorchをインストールした。
その後、import torch
で下記エラーが出てきた。
OSError: [WinError 126] fbgemm.dll" or one of its dependencies
解決方法
この記事を参考にして、torchらのバージョンを2.4.0から2.3.1にダウングレードしたらうまくいった。
# 一度アンインストール
pip uninstall torch torchvision torchaudio
# バージョンを指定してインストール
pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1
原因(pytorchのバージョン)
pytorch 2.4.0
は python 3.12
でも使えると書いてあったが、自分の環境ではできなかった。(仮想環境を作り直してもダメ)
別の要因
PyTorch は、いくつかの依存関係として Microsoft Visual C++ ランタイムを必要とする。これがインストールされていないか、バージョンが古い場合、DLL のロードに失敗することがある。