Closed3
[Python / YOLOv8] GPU使って学習させたいってワケ
あざらしnvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Thu_Mar_28_02:30:10_Pacific_Daylight_Time_2024
Cuda compilation tools, release 12.4, V12.4.131
Build cuda_12.4.r12.4/compiler.34097967_0
を使ってモデル作成したーい
あざらしpipで入れるもの
普通に入れる
- ultralytics
- labelimg
- pycuda
バージョン指定したほうがいい
- torch
- torchvision
- torchaudio
以下は例
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
公式みてすぺっくなどいろいろ指定するとコマンド出してくれるよ
バージョンが合っていないと下記エラーでるよ
インストーラーから入れるもの
CUDA Toolkit
すぺっくに合わせてインスコする
Nsight Computeでエラーを吐く場合
カスタムインスコにして除外すればおｋ
GPUドライバ
こっちもあわせてインスコ
※バージョンアップの場合もここからインスコしてあげるのがはやいらしい
C++ビルドツール
普通にインスコ
あざらしあとはトレインモードで下記のようにして実行する
model.train(data=DATASETS, epochs=100, device="cuda")
このスクラップは2024/04/24にクローズされました