深度学习多卡训练

77 阅读 0 评论 51 点赞

我是靠谱客的博主拼搏斑马，最近开发中收集的这篇文章主要介绍深度学习多卡训练，觉得挺不错的，现在分享给大家，希望可以做个参考。

os.environ['CUDA_VISIBILE_DEVICE']='4,5,6,7'

net = AttU_Net().to("cuda")
    
net = nn.DataParallel(net, device_ids=[0,1,2,3])

直接运行或者是在终端输入下面的命令：

CUDA_VISIBLE_DEVICES=4,5,6,7 python test.py

以上就是拼搏斑马为你收集整理的深度学习多卡训练的全部内容，希望文章能够帮你解决深度学习多卡训练所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。