microsoft / GLIP

Grounded Language-Image Pre-training
MIT License
2.07k stars 186 forks source link

the docker doesn't support 4090 GPU #144

Open whhong5 opened 9 months ago

whhong5 commented 9 months ago

Great work! I tried to pull the docker image "pengchuanzhang/pytorch:ubuntu20.04_torch1.9-cuda11.3-nccl2.9.9" but when I run this image, I got the error message shown blow:

"WARNING: Detected NVIDIA NVIDIA GeForce RTX 4090 GPU, which is not yet supported in this version of the container"

I don't know how to solve this problem. Could you please give me some suggestions?

ybk66 commented 9 months ago

可能是4090太先进了?

liunian-harold-li commented 8 months ago

Yeah, 4090 might be too advanced for our pre-built docker. You might need to manually install the dependencies.