PyTorch学习系列(十六)——如何使用cuda进行训练？
2017-06-05 21:46
585 查看
如果想在CUDA上进行计算，需要将操作对象放在GPU内存中。
对于普通的张量，可以直接：
对于神经网络：
同一个GPU上的张量计算结果仍然保存在该GPU上。
对于普通的张量，可以直接：
x = torch.randn(2, 3) x = x.cuda()
对于神经网络：
model = MyModel() model.cuda()
同一个GPU上的张量计算结果仍然保存在该GPU上。
参考
[1] http://pytorch.org/docs/notes/cuda.html#
内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理 
相关文章推荐
- 如何在k8s集群中安装nvidia、cuda并使用GPU进行训练
- 在cuda8.0+faster-rcnn（python版）下使用kitti数据集进行训练
- PyTorch学习系列(十)——如何在训练时固定一些层？
- Linux 网络渗透 如何使用Aircrack-ng 系列工具进行WPA/WPA2的监听和破解
- 系列 《使用sklearn进行集成学习——理论》 《使用sklearn进行集成学习——实践》 目录 1 Random Forest和Gradient Tree Boosting参数详解 2 如何调参？
- Zookeeper系列三：Zookeeper客户端的使用（Zookeeper原生API如何进行调用、ZKClient、Curator）
- django “如何”系列1：如何使用REMOTE_USER（远程用户）进行认证
- 【量化小讲堂-Python&Pandas系列01】如何快速上手使用Python进行金融数据分析
- Kaldi中如何使用已经训练好的模型进行语音识别ASR呢？
- 如何使用word2vec进行分词训练
- 如何使用u 4000 nity3d进行地形编辑/百分之一系列<1>
- nw.js node-webkit系列（15）如何使用内部模块和第三方模块进行开发
- 【玩转单片机系列002】 如何使用STM32提供的DSP库进行FFT
- PyTorch学习系列(十五)——如何加载预训练模型？
- 如何使用圆体来美化,支持中文粗体和斜体(Ubuntu)? 使用xp带的字体来进行中文美化
- Enterprise Library Step By Step系列（十六）：使用AppSetting Application Block
- 如何使用自定义表单进行数据安全性配置
- [How To]如何使用Wildfish进行ISeries项目开发--数据库建立部分
- 数据绑定技术—如何使用DataBinder.Eval()方法进行数据绑定
- 如何使用Lua扩展C/C++应用系列1