PyTorch torch.set_default_device 関数
Pytorch torch リファレンスマニュアルtorch.set_default_devicePyTorch でデフォルトデバイスを設定するための関数です。
関数定義
torch.set_default_device(device)
使用例
実例
import torch
# デフォルトデバイスを設定
if torch.cuda.is_available():
torch.set_default_device('cuda')
x = torch.zeros(3, 4)
print("デフォルトデバイス:", x.device)
torch.set_default_device('cpu')
else:
print("CUDA 利用不可")
# デフォルトデバイスを設定
if torch.cuda.is_available():
torch.set_default_device('cuda')
x = torch.zeros(3, 4)
print("デフォルトデバイス:", x.device)
torch.set_default_device('cpu')
else:
print("CUDA 利用不可")
その他の拡張