W&B Launch を GPU 上での Tensorflow と連携させるにはどうすればよいですか?
less than a minute
TensorFlow ジョブで GPU を使用する場合、コンテナビルド用にカスタムベースイメージを指定します。これにより、run 中の正しい GPU 利用が保証されます。リソース設定の builder.accelerator.base_image
キーの下にイメージタグを追加します。例えば:
{
"gpus": "all",
"builder": {
"accelerator": {
"base_image": "tensorflow/tensorflow:latest-gpu"
}
}
}
W&B バージョン 0.15.6 以前では、base_image
の親キーとして accelerator
の代わりに cuda
を使用してください。
フィードバック
このページは役に立ちましたか?
Glad to hear it! If you have further feedback, please let us know.
Sorry to hear that. Please tell us how we can improve.