随笔分类 -  PyTorch

Pytorch：使用GPU训练
摘要：1.模型转为cuda 2.数据转为cuda 3.输出数据去cuda，转为numpy linux输入nvidia smi，可以看到调用GPU成功！ 阅读全文

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

理解PyTorch的自动微分机制
摘要：参考 "Getting Started with PyTorch Part 1: Understanding how Automatic Differentiation works" 非常好的文章，讲解的非常细致。 注意这篇文章基于v0.3，其中的Variable和Tensor在后来把版本中已经合并 阅读全文

posted @ 2019-05-11 10:43 Frank_Allen 阅读(736) 评论(0) 推荐(0)

导航