some problem about running LAPGAN

witnessai / LAPGAN

Unofficial PyTorch implementation of LAPGAN (Deep Generative Image Models using a Laplacian Pyramid of Adversarial Networks, NIPS 2015)

14 stars 8 forks source link

some problem about running LAPGAN #1

Open zsyai799 opened 4 years ago

zsyai799 commented 4 years ago

RuntimeError: cuda runtime error (100) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cpp:50 When I run train_lapgan.py in colaboratory,there is always the error. If you know how to solve this problem, please tell me. Thanks.

xxxnb commented 3 years ago

I have the same problem

CYummy0312 commented 2 years ago

Maybe you should set the right number of cuda device, like os.environ['CUDA_VISIBLE_DEVICES'] = '0' ?