PyTorch torch.cuda.is_available 関数
Pytorch torch リファレンスマニュアルtorch.cuda.is_availablePyTorch で CUDA が利用可能かどうかを確認するための関数です。
関数定義
torch.cuda.is_available()
使用例
例
import torch
# CUDA が利用可能かどうかを確認
if torch.cuda.is_available():
print("CUDA 利用可能!")
print("GPU 数:", torch.cuda.device_count())
print("GPU 名称:", torch.cuda.get_device_name(0))
else:
print("CUDA 利用不可")
# CUDA が利用可能かどうかを確認
if torch.cuda.is_available():
print("CUDA 利用可能!")
print("GPU 数:", torch.cuda.device_count())
print("GPU 名称:", torch.cuda.get_device_name(0))
else:
print("CUDA 利用不可")
その他の拡張