|名字 |功能 |参数 |定义 |备注1 |备注2
|—- |—- |—- |—- |—- |—-
|torch.cuda.is_available()| 查看gpu是否可用
|torch.cuda.device_count()| gpu数量
|torch.cuda.current_device()| 当前gpu索引号
|torch.cuda.get_device_name()| 根据gpu 索引号查看gpu名字 输出 ‘GeForce GTX 1050’ | gpu的index 从0开始
上一篇
Pytorch函数总结-模型相关函数
2021-09-15
下一篇
Pytorch函数总结-常用函数
2021-09-14