search

zzzyuqing / DreamMat

[SIGGRAPH2024] DreamMat: High-quality PBR Material Generation with Geometry- and Light-aware Diffusion Models
MIT License
246 stars 9 forks source link

CUDA not found in docker image #14

Closed mlong93 closed 2 days ago

mlong93 commented 2 weeks ago

I pulled the official docker image but after installing all the dependencies, I'm getting torch.cuda.is_available() = False.

Do you have further instructions to use the Docker image?

XZYW7 commented 2 weeks ago

I encountered the same problem and the command nvidia-smi didn’t work in docker. I solved this by installing the NVIDIA Container Toolkit and adding the parameter --gpus all when running docker. If you haven’t installed it, you can refer to this issue, and the installation guide