PyTorch Memo 備忘録

1. GPU確認コード

00.torch_gpu.py

import torch

print("PyTorch version:", torch.__version__)

if torch.cuda.is_available():
    device = torch.device("cuda")          # a CUDA device object
    print(f"Using {torch.cuda.get_device_name()} for PyTorch")
else:
    device = torch.device("cpu")           # a CPU object
    print("No GPU available, using CPU for PyTorch")

result

PyTorch version: 2.0.1+cu118
Using NVIDIA GeForce RTX 3060 Laptop GPU for PyTorch