PyTorch torch.get_device_module 関数

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.get_device_moduleこれは、PyTorch で現在のデバイスモジュールを取得するために使用される関数です。

関数定義

torch.get_device_module(device='cuda')

使用例

実例

import torch

# CUDAデバイスモジュールを取得
cuda_module = torch.get_device_module('cuda')
print("CUDAモジュール:", cuda_module)

# CUDAが利用可能かどうかを確認
print("CUDA利用可能:", torch.cuda.is_available())

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張