JeffreyXiang / ivid

PyTorch implementation of the ICCV paper "3D-aware Image Generation using 2D Diffusion Models"
https://jeffreyxiang.github.io/ivid/
MIT License
300 stars 4 forks source link

Exception: requested device index 0, but found 0 devices #11

Open Zhangkaiwen2001 opened 5 months ago

Zhangkaiwen2001 commented 5 months ago

Traceback (most recent call last): File "/data/zhangkaiwen/ivid-main/inference/sample.py", line 342, in main(0, 1, seeds, modelviews, classes, cfg, cfg_uncond, cfg_cond) File "/data/zhangkaiwen/ivid-main/inference/sample.py", line 223, in main for i, (meshes, colors, samples, conds) in enumerate(sample_process): File "/data/zhangkaiwen/anaconda3/envs/I2V_attack/lib/python3.9/site-packages/torch/utils/_contextlib.py", line 35, in generator_context response = gen.send(None) File "/data/zhangkaiwen/ivid-main/inference/sample.py", line 50, in sample_all renderer = [rgbd3d.AggregationRenderer(128 * 3, 128) for in range(batchsize)] File "/data/zhangkaiwen/ivid-main/inference/sample.py", line 50, in renderer = [rgbd3d.AggregationRenderer(128 * 3, 128) for in range(batchsize)] File "/data/zhangkaiwen/ivid-main/rgbd_3d/moderngl_renderer.py", line 169, in init self.init_gl_context(device) File "/data/zhangkaiwen/ivid-main/rgbd_3d/moderngl_renderer.py", line 177, in init_gl_context self.mgl_ctx = moderngl.create_context(standalone=True, backend='egl', device_index=device) File "/data/zhangkaiwen/anaconda3/envs/I2V_attack/lib/python3.9/site-packages/moderngl/init.py", line 1936, in create_context ctx.mglo, ctx.version_code = mgl.create_context(glversion=require, mode=mode, settings) File "/data/zhangkaiwen/anaconda3/envs/I2V_attack/lib/python3.9/site-packages/glcontext/init.py", line 120, in create return egl.create_context(kwargs) Exception: requested device index 0, but found 0 devices Exception ignored in: <function AggregationRenderer.del at 0x7f370ed44280> Traceback (most recent call last): File "/data/zhangkaiwen/ivid-main/rgbd_3d/moderngl_renderer.py", line 246, in del self.mgl_ctx.release() AttributeError: 'NoneType' object has no attribute 'release'

can you tell me what should i do?

Zhangkaiwen2001 commented 5 months ago

command: CUDA_VISIBLE_DEVICES=2 python inference/sample.py --num_samples 10000 --classes uniform --viewset 3x9 --steps_uncond 1000 --steps_cond 50 --guidance 0.5

ewrfcas commented 2 months ago

Some GPUs are not supported.