2019-01-11から1日間の記事一覧

PyTorch C++ APIでGPUを使ってみる

今回はPyTorchのC++ APIでGPUを使った学習と推論を試してみる. (adsbygoogle = window.adsbygoogle || []).push({}); GPUの使い方 C++でGPUが使えるかどうかは,以下のように確認できる torch::cuda::is_available() ここはpythonとほとんど変わらない GPU…