2020年7月9日
【PyTorch】使用笔记
摘要： 检查使用的GPU的索引 torch.cuda.is_available() cuda是否可用； torch.cuda.device_count() 返回gpu数量； torch.cuda.get_device_name(0) 返回gpu名字，设备索引默认从0开始； torch.cuda.curren 阅读全文
posted @ 2020-07-09 14:28 YeZzz 阅读(288) 评论(0) 推荐(0) 编辑