2024年8月6日
pytorch张量运算
摘要： pytorch张量运算 使用cpu gpu训练数据 torch.cuda.is_available() # 判断机器支不支持cpu 一般用如下命令: device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") fm 阅读全文
posted @ 2024-08-06 22:09 金字塔下的蜗牛 阅读(50) 评论(0) 推荐(0) 编辑