- Windows WSL内でGPUを使えるようにしたはず
- Nxで使いたい
- という状態で、どう確認を取ればいいのかわからない
結論
まずは、WSL上でのインストール確認
$ nvidia-smi上記で出力があれば問題ない様子
次に、docker環境下でGPUが見えていることを確認。これは上記コマンドは何かしらのコンテナで実行する
$ docker run -it --rm --gpus all elixir:1.14 /bin/bash
# nvidia-smi最後に、Livebook等でNxをGPUで使えるか確認するコードを書く。エラーがでないことと速度が速いこと
Nxの中にあるベンチマーク用のソースを参考に見比べる
https://github.com/elixir-nx/nx/blob/main/exla/bench/softmax.exs
EXLA_TARGET="cuda"を動かすにはビルドが必要だったりでわりと準備が多い様子- さくっと試すならばexlaで提供されているDockerfileを使うのが良さそう
経過
(RuntimeError) Could not find registered platform with name: "cuda". Available platform names are: Host