2023-01-01から1ヶ月間の記事一覧

DirectML使ってみた

冬は寒いのでDNNの学習を回すのにぴったり!GPUの廃熱で暖房費節約だぜ!などと思ったけれど、メインマシンはメインOSはWindowsで運用していてビデオカードもAMDなのでDNNフレームワークを動かすのはしんどい。 調べたらDirectMLってのでWindows+AMDでもいけ…

venvのトップディレクトリにPyTorchの重みファイル(pth)を置いちゃダメって話

横着したらハマったので自戒を込めてメモ。 まとめ Pythonには「サイト固有の設定フック」という機能がある venv環境のトップディレクトリに入っている .pth ファイルを読み込んで処理する PyTorchの学習済み重みファイル(拡張子 .pth)を置いていると上記…

CUDA実装のONNXRuntimeカスタムオペレータを実装してみた

前回の続き。 前回の記事 ↓ maminus.hatenadiary.org 今回のソースコード ↓ github.com ありがたいことにGitHubのIssueでCUDA版のカスタムオペレータ実装方法について問い合わせをいただいたので時間ができた時に実装してみた。 ※ONNXRuntimeのカスタムオペ…