問題タブ [azureml-python-sdk]
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.
90 問題
0
投票する
1
に答える
79
参照
azure - PyTorch で CUDA を使用しない AzureML 実験パイプライン
PyTorch と CUDA を使用してモデルをトレーニングする実験パイプラインを実行しています。以下のように環境を作りました。
そして、ここにトレーニングステップがあります：
トレーニング スクリプトを実行するときにマシンを使用しているにもかかわらずStandard_NC12_Promo、GPU が PyTorch によって取得されませんdevice = torch.device("cuda" if torch.cuda.is_available() else "cpu")。
実験ではなく同じマシンでスクリプトを実行しようとすると、GPU が使用されます。
これに対する潜在的な解決策を知っていますか？