S_unn_y_ 个人主页

@S_unn_y_

S_unn_y_

2023-07-21 20:29:32 加入 DevPress

简介

该用户还未填写简介

擅长的技术栈

未填写擅长的技术栈

可提供的服务

暂无可提供的服务

是否使用gpu的判断方法

print("GPU名称：", torch.cuda.get_device_name(1))# 根据索引号得到GPU名称。# print("是否可用：", torch.cuda.is_available())# 查看GPU是否可用。# print("GPU数量：", torch.cuda.device_count())# 查看GPU数量。# print("GPU索引号：", torch.cuda.

#python #pytorch #深度学习

到底了