PyTorch torch.device 関数
Pytorch torch リファレンスマニュアルtorch.devicePyTorchでデバイスオブジェクトを作成するための関数であり、テンソルを格納するデバイスを指定するために使用します。
関数定義
torch.device(device)
使用例
例
import torch
# CPUデバイスを作成
cpu_device = torch.device('cpu')
print("CPUデバイス:", cpu_device)
# CUDAデバイスを作成
if torch.cuda.is_available():
cuda_device = torch.device('cuda:0')
print("CUDAデバイス:", cuda_device)
# GPU上にテンソルを作成
x = torch.zeros(3, 4, device=cuda_device)
print("テンソルデバイス:", x.device)
# CPUデバイスを作成
cpu_device = torch.device('cpu')
print("CPUデバイス:", cpu_device)
# CUDAデバイスを作成
if torch.cuda.is_available():
cuda_device = torch.device('cuda:0')
print("CUDAデバイス:", cuda_device)
# GPU上にテンソルを作成
x = torch.zeros(3, 4, device=cuda_device)
print("テンソルデバイス:", x.device)
その他の拡張