PyTorch torch.is_available 関数
Pytorch torch リファレンスマニュアルtorch.is_availablePyTorch で CUDA が利用可能かどうかを確認するための関数です。True が返された場合は CUDA が利用可能、False の場合は利用不可を意味します。
関数定義
torch.is_available()
使用例
例
import torch
# CUDA が利用可能かどうかを確認
if torch.is_available():
print("CUDA is available!")
print(f"CUDA device count: {torch.cuda.device_count()}")
else:
print("CUDA is not available")
# CUDA が利用可能かどうかを確認
if torch.is_available():
print("CUDA is available!")
print(f"CUDA device count: {torch.cuda.device_count()}")
else:
print("CUDA is not available")
その他の拡張