PyTorch torch.cuda 関数
Pytorch torch リファレンスマニュアルtorch.cudaPyTorch で CUDA を管理するためのモジュールです。CUDA デバイス管理、メモリ管理、同期などの機能を提供します。
関数定義
torch.cuda
torch.cuda モジュールには、以下の一般的な関数が含まれています:
torch.cuda.is_available()- CUDA が利用可能かどうかを確認するtorch.cuda.device_count()- CUDA デバイスの数を取得するtorch.cuda.current_device()- 現在のデバイスを取得するtorch.cuda.synchronize()- CUDA 操作を同期するtorch.cuda.empty_cache()- キャッシュを空にする
使用例
実例
import torch
# CUDA が利用可能かどうかを確認する
print(f"CUDA available: {torch.cuda.is_available()}")
# CUDA デバイスの数を取得する
if torch.cuda.is_available():
print(f"CUDA device count: {torch.cuda.device_count()}")
print(f"CUDA device name: {torch.cuda.get_device_name(0)}")
# CUDA が利用可能かどうかを確認する
print(f"CUDA available: {torch.cuda.is_available()}")
# CUDA デバイスの数を取得する
if torch.cuda.is_available():
print(f"CUDA device count: {torch.cuda.device_count()}")
print(f"CUDA device name: {torch.cuda.get_device_name(0)}")
その他の拡張