コンピュータに利用可能なGPUデバイスがあるかどうかを確認する
2022-02-16 19:57:48
コード
def is_gpu_available(cuda_only=True):
from tensorflow.python.client import device_lib as _device_lib
if cuda_only:
return any((x.device_type == 'GPU')
for x in _device_lib.list_local_devices())
else:
return any((x.device_type == 'GPU' or x.device_type == 'SYCL')
for x in _device_lib.list_local_devices())
print(is_gpu_available(cuda_only=True))
2020-11-29 15:55:26.179560: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
2020-11-29 15:55:26.453227: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1432] Found device 0 with properties:
name: GeForce GTX 1650 major: 7 minor: 5 memoryClockRate(GHz): 1.56
pciBusID: 0000:01:00.0
totalMemory: 4.00GiB freeMemory: 3.25GiB
2020-11-29 15:55:26.453414: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1511] Adding visible gpu devices: 0
2020-11-29 15:55:27.273123: I tensorflow/core/common_runtime/gpu/gpu_device.cc:982] Device interconnect StreamExecutor with strength 1 edge matrix:
2020-11-29 15:55:27.273240: I tensorflow/core/common_runtime/gpu/gpu_device.cc:988] 0
2020-11-29 15:55:27.273301: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1001] 0: N
2020-11-29 15:55:27.273504: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1115] Created TensorFlow device (/device:GPU:0 with 2941 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1650, pci bus id: 0000:01:00.0, compute capability: 7.5)
True
結果表示
2020-11-29 15:55:26.179560: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
2020-11-29 15:55:26.453227: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1432] Found device 0 with properties:
name: GeForce GTX 1650 major: 7 minor: 5 memoryClockRate(GHz): 1.56
pciBusID: 0000:01:00.0
totalMemory: 4.00GiB freeMemory: 3.25GiB
2020-11-29 15:55:26.453414: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1511] Adding visible gpu devices: 0
2020-11-29 15:55:27.273123: I tensorflow/core/common_runtime/gpu/gpu_device.cc:982] Device interconnect StreamExecutor with strength 1 edge matrix:
2020-11-29 15:55:27.273240: I tensorflow/core/common_runtime/gpu/gpu_device.cc:988] 0
2020-11-29 15:55:27.273301: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1001] 0: N
2020-11-29 15:55:27.273504: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1115] Created TensorFlow device (/device:GPU:0 with 2941 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1650, pci bus id: 0000:01:00.0, compute capability: 7.5)
True
関連
-
SSHパスワードフリーログイン設定後、stdinがターミナルでないため、疑似ターミナルが割り当てられない。
-
Linux の ssh 公開鍵の追加
-
アクセスが拒否されました。PROCESS 特権が(少なくとも1つ)必要です。
-
時間問題をたくさん出してくれた人たち(連続更新)
-
R パッケージのインストール: パッケージ 'phangorn' のインストールで終了ステータスが 0 以外になった。
-
LDAPエラー ldap_bind: 無効な認証情報 (49) 解決策
-
Ubuntu next install nginx,make: *** `default'が必要とする `build' をターゲットとするルールがありません。停止します。
-
ubuntuでグローバルシステムエージェントを設定する
-
Linuxでビジー状態のテキストファイルでプログラムを実行する場合の対処法
-
pip: コマンドが見つかりません
最新
-
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 実装 サイバーパンク風ボタン
おすすめ
-
undefinedmakefile のエラー *** セパレータがありません。
-
undefinedmakefile:n: *** セパレータがありません。
-
psycopgのアセンブルはエラーになります。Error: pg_config 実行ファイルが見つかりません。
-
Nvidia グラフィックスカード 初期化に失敗しました NVML Driver/library version mismatch エラー解決法
-
警告について:互換性のないポインタ型からの初期化【デフォルトで有効
-
PackagesNotFoundError: 以下のパッケージは、現在のチャネルから利用できません。
-
gcc/g++エラー... /crt1.o: 関数 `_start' で。(.text+0x18): `main'への未定義の参照エラー
-
Java Runtime Environmentを継続するためのメモリが不足しています。
-
Ansibleの通常ユーザーsudoがコマンドを実行します。
-
dpkg: パッケージの処理エラー ***(--install): 依存関係の問題 - 未設定にする