2018年9月3日
Pytorch --- cuda 相关
摘要： 说明: 本篇是对 Pytorch Docs 0.4.1 CUDA semeantics 的部分翻译和自己在阅读过程中想加入的一些注释。 1、torch.cuda torch.cuda 用来建立和运行 CUDA 操作。对于 Tensor 来说需要转化为 cuda tensor 来使用 cuda 计算。 阅读全文
posted @ 2018-09-03 14:15 赵小春 阅读(613) 评论(0) 推荐(0) 编辑