youzhijiangs
youzhijiangs
采纳率0%
2021-03-06 23:16

StyleGAN2多显卡报错

100

K80显卡 stylegan2encoder

只能用一个GPU跑正常 设置8个的话跑不了 num-gpus=1是下面的图完全没问题 num-gpus=8的话就不行了 跑着直接报错 跑的是1024x1024的      512x512的num-gpus=8 第一个0GPU下面别的卡显存上来后也报错 这样是不是因为K80单核心是12G显存太小了导致的? minibatch_gpu_base=2最小了都不行

有什么办法能让4个K80跑1024的吗? 你要说他跑不了1024但是你设置num-gpus=1就行 单卡能跑多卡跑不了奇葩求告知

又测试的GPU=2都不行只能1

这个是多卡报错图

 

  • 点赞
  • 收藏
  • 复制链接分享

2条回答

  • qq_28024621 Angel OvO 1月前
    启用8个GPU
    windows CMD:
    set CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
    linux:
    export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
    点赞 评论 复制链接分享
  • bill20100829 歇歇 1月前

    用7个核心

     

    点赞 评论 复制链接分享