問題タブ [azure-machine-learning-service]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
docker - NVIDIA CUDA カスタム Docker ベース イメージを使用して Azure ML で GPU を使用するにはどうすればよいですか?
カスタム Docker ベース イメージをビルドするための Dockerfile で、次のベース イメージを指定します。
nvidia-cuda ベース イメージに対応する dockerfile は、https ://gitlab.com/nvidia/container-images/cuda/blob/master/dist/ubuntu16.04/10.1/devel/cudnn7/Dockerfile にあります。
AzureML ログを印刷すると、次のようになります。
私は得る
しかし、CPUではなくGPUが欲しいです。私は何を間違っていますか？
編集：何が必要なのか正確にはわかりません。ただし、次の情報を提供できます。azureml.core VERSION は 1.0.57 です。compute_target は次の方法で定義されます。
実験は次の方法で実行されます。
yaml ファイルには以下が含まれます。
EDIT 2:use_gpu = Trueアップグレードを試みましazureml-sdk=1.0.65たが、役に立ちませんでした。を介して cuda-drivers を追加でインストールすることを提案する人もapt-get install cuda-driversいますが、これは機能せず、docker イメージをビルドできません。nvcc --versiondocker イメージの出力は次のようになります。
それでいいと思いますが、もちろんdockerイメージ自体にはGPUがないので、コマンドnvidia-smiが見つからず、
その後
False を出力します。