DockerでGPU版ONNXを使ってみる
1k{icon} {views} DockerでGPU版をONNXを動かしてみました。比較的に簡単に動かせたので、いろいろ便利だと思います。YOLOXを例に検証します。 概要 ONNXRuntimeのGPU版をDocke […]...
FastAPI+MessagePackで画像とメタデータを一緒にPOSTする
668{icon} {views} FastAPIで画像とメタデータ(テキストやリストなど)を一緒に、1つのオブジェクトとしてPOSTする方法を解説します。MessagePackで全部丸々固めてしまうやり方で、サクッと作 […]...
Windows11+WSL2でGPUのPyTorchを最低限動かすためのオレオレDockerfile
2.2k{icon} {views} Windows11+WSL2でGPU対応のPyTorchのDockerイメージを1から作る方法を紹介します。CUDAとCuDNNはDockerhubで公開されているnvidiaのイメ […]...
Windows11でWSL2+nvidia-dockerでPyTorchを動かすのがすごすぎた
21.5k{icon} {views} Windows11にはWSL2があり、Linux環境からGPUありのPyTorchを動かすことが可能です。WSL2経由のほうがWindowsネイティブよりも訓練速度が倍になります。 […]...