2019年7月18日

Pytorch:使用GPU训练

摘要: 1.模型转为cuda 2.数据转为cuda 3.输出数据去cuda,转为numpy linux输入nvidia smi,可以看到调用GPU成功! 阅读全文

posted @ 2019-07-18 10:32 Frank_Allen 阅读(1607) 评论(0) 推荐(0) 编辑

导航