自定义博客皮肤VIP专享

*博客头图：

点击选择上传的图片

格式为PNG、JPG，宽度*高度大于1920*100像素，不超过2MB，主视觉建议放在右侧，请参照线上博客头图

请上传大于1920*100像素的图片！

博客底图：

点击选择上传的图片

图片格式为PNG、JPG，不超过1MB，可上下左右平铺至整个背景

栏目图：

点击选择上传的图片

图片格式为PNG、JPG，图片宽度*高度为300*38像素，不超过0.5MB

主标题颜色：

RGB颜色，例如：#AFAFAF

Hover：

RGB颜色，例如：#AFAFAF

副标题颜色：

RGB颜色，例如：#AFAFAF

预览取消提交

自定义博客皮肤

-+

上一步保存

Exploer_TRY的博客

Pytorch

关注

关注数：文章数：1 文章阅读量：222 文章收藏量：0

作者: CExploer

智慧、创造、奇迹

展开

Pytorch一些小问题记录

1. 单机多卡与单卡加载多卡问题训练单机多卡模型，很容易，先定义需要使用的显卡编号：import osos.environ["CUDA_VISIBLE_DEVICES"] = args.gpu_device # such as [0,2,3]加载模型：# GPUif torch.cuda.is_available(): model = model.cuda() # Multi GPU if torch.cuda.device_count() > 1:

原创 2021-04-11 13:50:04 · 223 阅读 · 0 评论