pytorch多GPU使用的方法model=nn.DataParallel(model).cuda()

2022-10-24 09:06:18

单GPU：

importosos.environ["CUDA_VISIBLE_DEVICES"] ="0"
  • 1
  • 2

多GPU：

device_ids= [0,1,2,3]

单GPU：

importosos.environ["CUDA_VISIBLE_DEVICES"] ="0"
  • 1
  • 2

多GPU：

device_ids= [0,1,2,3]
  • 作者：AIchiNiurou
  • 原文链接：https://blog.csdn.net/weixin_44523062/article/details/108945103
    更新时间：2022-10-24 09:06:18