文章目录 一、需要使用cuda的对象 二、使用GPU训练方法 1. .cuda() 2. .to(device) 一、需要使用cuda的对象 数据data,模型model,损失loss 二、使用GPU训练方法 1. .cuda() os.environ['CUDA_VISIBLE_DEVICE']='0' # 指定0号GPU if torch.cuda.is_a