m0_54644067的博客

目录一、Pytorch中怎么调用cuda？？1. 第一种2. 第二种二、哪些对象可以使用cuda?1.网络模型 (module)2.数据(输入imgs、标注target/label)3.损失函数 (optimizer)一、Pytorch中怎么调用cuda？1. 第一种if torch.cuda.is_available(): my_module = my_module.cuda()2. 第二种# 定义设备device = ...