![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
机器学习
fjy2035
这个作者很懒,什么都没留下…
展开
-
CSDN-新手第一课:如何转载别人的博客
注意:转载别人的HTML代码时,在复制到CSDN–markdown编辑器后,为了整个页面的布局不留大量空白,首尾代码可删除。 参考博客:https://blog.csdn.net/zhongjianblackberry/article/details/79456338 转载CSDN博客步骤: 1.CSDN博客页面右键,点击【检查】 点击检查后,页面右侧出现html代码,如下图 2.如果需要转载全文,则在html代码下侧点击选中article_content 即可,会在代码框中自转载 2020-10-21 17:54:08 · 201 阅读 · 0 评论 -
PyTorch指定单GPU和多GPU训练及保存-加载模型(含CPU)的总结
注意:训练/测试过程中 inputs 和 labels 均需加载到GPU中。参考和综合了网上的多种方法。1. PyTorch使用指定GPU训练 - 单GPU 直接使用代码 model.cuda(), PyTorch默认从0开始的单GPU:model = Model()if torch.cuda.is_available(): model = model.cuda()有两种方法可直接指定单GPU:在终端shell:CUDA_VISIBLE_DEVICES=1 python main.p原创 2020-10-21 19:31:12 · 13118 阅读 · 7 评论