2022年7月6日
检测是否torch的CUDA是否可用的代码
摘要： import torch print(torch.cuda.is_available()) # 返回True 接着用下列代码进一步测试 print(torch.zeros(1).cuda()) 阅读全文
posted @ 2022-07-06 17:13 乌蝇哥 阅读(582) 评论(0) 推荐(0) 编辑