[解決済み] CUDAをpytorchで使う?
質問内容
このサイトで検索しましたが、古い投稿しか見つかりませんでした。
GPUでトレーニングを実行したいのですが、どうすればいいですか?いくつかのフォーラムで、私は私が適用する必要があることを発見しました。
.cuda()
CUDAを使いたいものすべてに適用しました(私はプログラムがクラッシュしないように、できる限りのものに適用しました)。驚いたことに、これではトレーニングがさらに遅くなってしまいます。
すると、こんな使い方があることがわかりました。
torch.set_default_tensor_type('torch.cuda.FloatTensor')
を使うとCUDAが使えるようになります。両方有効にしても何も変わりません。何が起きているのでしょうか?
モデル全体で確実にCUDAを有効にする方法はありますか?
また
MyModel()
とはどういう意味ですか?もっと具体的な例、例えばコード例が欲しい。
(これは私が参照している投稿です)
解決方法は?
を使用することができます。
tensor.to(device)
コマンドを使用して、テンソルをデバイスに移動する。
は、その
.to()
コマンドは、リンク先の投稿のように、モデル全体をデバイスに移動する場合にも使用されます。
もう一つの可能性は、テンソルの作成時に、テンソルのデバイスを設定するために
device=
のようにキーワード引数で指定します。
t = torch.tensor(some_list, device=device)
コード内で動的にデバイスを設定したい場合は
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
を使用して、可能であればcudaをデバイスとして設定します。
に様々なコード例があります。 PyTorchチュートリアル や上記のリンク先のドキュメントが参考になると思います。
関連
-
[解決済み】RuntimeWarning: invalid value encountered in double_scalars で numpy の除算ができない。
-
[解決済み】SyntaxError: デフォルト以外の引数がデフォルトの引数に続く
-
[解決済み] PyTorchの "view "メソッドはどのように動作するのですか?
-
[解決済み] 関数内でグローバル変数を使用する
-
[解決済み] pipでPythonの全パッケージをアップグレードする方法
-
[解決済み] モジュールの関数名(文字列)を使って、モジュールの関数を呼び出す。
-
[解決済み] pipで特定のバージョンのパッケージをインストールする
-
[解決済み] pytorchがGPUを使用しているかどうかを確認するにはどうすればよいですか?
-
[解決済み】forループを使った辞書の反復処理
-
[解決済み】大文字と数字を含むランダムな文字列の生成
最新
-
nginxです。[emerg] 0.0.0.0:80 への bind() に失敗しました (98: アドレスは既に使用中です)
-
htmlページでギリシャ文字を使うには
-
ピュアhtml+cssでの要素読み込み効果
-
純粋なhtml + cssで五輪を実現するサンプルコード
-
ナビゲーションバー・ドロップダウンメニューのHTML+CSSサンプルコード
-
タイピング効果を実現するピュアhtml+css
-
htmlの選択ボックスのプレースホルダー作成に関する質問
-
html css3 伸縮しない 画像表示効果
-
トップナビゲーションバーメニュー作成用HTML+CSS
-
html+css 実装 サイバーパンク風ボタン
おすすめ
-
PythonによるLeNetネットワークモデルの学習と予測
-
opencvとpillowを用いた顔認証システム(デモあり)
-
pyCaret効率化乗算器 オープンソース ローコード Python機械学習ツール
-
Python入門 openを使ったファイルの読み書きの方法
-
Pythonの@decoratorsについてまとめてみました。
-
[解決済み】Python regex AttributeError: 'NoneType' オブジェクトに 'group' 属性がない。
-
[解決済み】ImportError: PILという名前のモジュールがない
-
[解決済み】TypeError: re.findall()でバイトのようなオブジェクトに文字列パターンを使用することはできません。)
-
[解決済み】"No JSON object could be decoded "よりも良いエラーメッセージを表示する。
-
[解決済み】cアンダースコア式`c_`は、具体的に何をするのですか?