pytorch基础

安装torch2的cu118版本
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

检查是否已有cuda合并列表，如果有可用cuda默认设置为cuda

import torch

if torch.cuda.is_available():
    device = torch.device("cuda")          # a CUDA device object
    print("已使用cuda")
else:
    device = torch.device("cpu") #设置默认使用cpu
    print("已使用 CPU")

或者按这个来检测是否cpu和cuda
cuda = device.type != 'cpu'

打开工具箱 dir()
获得帮助 help()