本文介绍了如何使用CUDA进行计算的基本原则：确保所有输入数据及网络模型均为CUDA张量形式；并详细说明了如何将普通张量转换为CUDA张量的方法。
1/ When you are creating a target of training samples, it must be a tensor.
2/ If you want to use cuda to compute, the data that input the net:forward/backward and criterion forward/backward must be cuda tensor. And net and criterion can use net:cuda() directly to become the net for cuda. However, data like tensor must be transformed by data = data:cuda(). Keep in mind.
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
