GPU で curve fitting する gpufit のメモ
777 ワード
GPU(CUDA) で Levenberg Marquardt 法(LM 法)でフィッティング(最適化する)ライブラリです.
Windows はバイナリがありますが, Linux はソースコードからビルドが必要です.
python binding
python バインディングもあります
(cmake ビルド時に whl が生成される)
git repo のサンプルコードみればだいたい使い方はわかるかと思います(Numpy を使っています).
注意事項
最新(CUDA 10 くらい?)nvcc だと sm30 などの古いアーキテクチャはサポートされなくなったため, cmake そのままですとアーキテクチャ選択がうまくいかず sm30 など古いアーキテクチャも含めてしまうため, コンパイルがコケます.
-DCUDA_ARCHITECTURES="7.0"
あたりで利用するアーキテクチャを明示的に cmake に指定しましょう.
Author And Source
この問題について(GPU で curve fitting する gpufit のメモ), 我々は、より多くの情報をここで見つけました https://qiita.com/syoyo/items/cfc4032ee0c5a84ac856著者帰属:元の著者の情報は、元のURLに含まれています。著作権は原作者に属する。
Content is automatically searched and collected through network algorithms . If there is a violation . Please contact us . We will adjust (correct author information ,or delete content ) as soon as possible .