Can't work on device cuda:1 - Githubissues

open-mmlab / mmgeneration

MMGeneration is a powerful toolkit for generative models, based on PyTorch and MMCV.

https://mmgeneration.readthedocs.io/en/latest/

Apache License 2.0

1.89k stars 227 forks source link

Can't work on device cuda:1 #500

Open rolfma19 opened 1 year ago

rolfma19 commented 1 year ago

python demo/unconditional_demo.py \ configs/styleganv2/stylegan2_c2_ffhq_256_b4x8_800k.py \ https://download.openmmlab.com/mmgen/stylegan2/stylegan2_c2_ffhq_256_b4x8_20210407_160709-7890ae1f.pth \ --save-path test.png \ --device cuda:0

test

python demo/unconditional_demo.py \ configs/styleganv2/stylegan2_c2_ffhq_256_b4x8_800k.py \ https://download.openmmlab.com/mmgen/stylegan2/stylegan2_c2_ffhq_256_b4x8_20210407_160709-7890ae1f.pth \ --save-path test.png \ --device cuda:1

test1