SyncDreamer icon indicating copy to clipboard operation
SyncDreamer copied to clipboard

Is there any docker?

Open r530044129 opened this issue 1 year ago • 11 comments

nice work! do you guys have any docker that we can pull? if have, it will be more convenient!

r530044129 avatar Sep 13 '23 08:09 r530044129

Hi! Thanks for the advice. I'll release a docker file for SyncDreamer.

liuyuan-pal avatar Sep 14 '23 04:09 liuyuan-pal

Hi, the docker is https://hub.docker.com/repository/docker/liuyuanpal/syncdreamer-env/general

liuyuan-pal avatar Sep 15 '23 12:09 liuyuan-pal

thank, i've pull this image, but there is something wrong when i run the python script, please see this cmd log

(base) [root /workspace/apple]$ python generate.py --ckpt ckpt/syncdreamer-pretrain.ckpt
> --input testset/aircraft.png \

               --output output/aircraft \
               --sample_num 4 \
               --cfg_scale 2.0 \
    >                    --elevation 30 \
               --crop_size 200

making attention of type 'vanilla' with 512 in_channels Working with z of shape (1, 4, 32, 32) = 4096 dimensions. making attention of type 'vanilla' with 512 in_channels Selected timesteps for ddim sampler: [ 1 6 11 16 21 26 31 36 41 46 51 56 61 66 71 76 81 86 91 96 101 106 111 116 121 126 131 136 141 146 151 156 161 166 171 176 181 186 191 196 201 206 211 216 221 226 231 236 241 246 251 256 261 266 271 276 281 286 291 296 301 306 311 316 321 326 331 336 341 346 351 356 361 366 371 376 381 386 391 396 401 406 411 416 421 426 431 436 441 446 451 456 461 466 471 476 481 486 491 496 501 506 511 516 521 526 531 536 541 546 551 556 561 566 571 576 581 586 591 596 601 606 611 616 621 626 631 636 641 646 651 656 661 666 671 676 681 686 691 696 701 706 711 716 721 726 731 736 741 746 751 756 761 766 771 776 781 786 791 796 801 806 811 816 821 826 831 836 841 846 851 856 861 866 871 876 881 886 891 896 901 906 911 916 921 926 931 936 941 946 951 956 961 966 971 976 981 986 991 996] loading model from ckpt/syncdreamer-pretrain.ckpt ... Traceback (most recent call last): File "generate.py", line 61, in main() File "generate.py", line 40, in main model = load_model(flags.cfg, flags.ckpt, strict=True) File "generate.py", line 19, in load_model model = model.cuda().eval() File "/root/miniconda3/lib/python3.8/site-packages/lightning_fabric/utilities/device_dtype_mixin.py", line 69, in cuda device = torch.device("cuda", torch.cuda.current_device()) File "/root/miniconda3/lib/python3.8/site-packages/torch/cuda/init.py", line 674, in current_device _lazy_init() File "/root/miniconda3/lib/python3.8/site-packages/torch/cuda/init.py", line 247, in _lazy_init torch._C._cuda_init() RuntimeError: No CUDA GPUs are available

And I use this cmd to start the docker: docker run -it --rm -v /mnt/d/Git/SyncDreamer:/workspace/apple liuyuanpal/syncdreamer-env

Is there any not correct?

r530044129 avatar Sep 18 '23 09:09 r530044129

Hi, you need to attach a GPU to the docker instance by --gpus.

liuyuan-pal avatar Sep 19 '23 02:09 liuyuan-pal

thanks, I've use this '--gpus all' to start the docker, but it's still has same error, torch.cuda.is_available() is false, is there some problem with the cuda version? image

r530044129 avatar Sep 21 '23 06:09 r530044129

Hi, I'm not sure about the problem. I'm able to load the GPU on my machine. image

Maybe, you could check whether you have GPU in this container by nvidia-smi.

liuyuan-pal avatar Sep 21 '23 07:09 liuyuan-pal

hi, could you please also provide Dockerfile? Thanks a lot!

JunyuanDeng avatar Sep 21 '23 09:09 JunyuanDeng

Hi, I generate the docker image from a container. I'm not sure about how to generate a docker file from it.

liuyuan-pal avatar Sep 21 '23 12:09 liuyuan-pal

It doesn't matter, thanks!

JunyuanDeng avatar Sep 22 '23 02:09 JunyuanDeng

Hi

I am trying to use the provided Docker image by pulling it and running a container, but it seems the requirements, including pip, are not installed in the Docker container.

I was wondering if that is the case, or if I am using the Docker incorrectly.

mshahbazi72 avatar Oct 12 '23 09:10 mshahbazi72

Hi, it should contain pip and python by default. Maybe, some bugs exist in using docker in your case.

liuyuan-pal avatar Oct 13 '23 06:10 liuyuan-pal