My CUDA Has seen the GPU but did not use it, why?

openai / CLIP

CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image

MIT License

26.1k stars 3.33k forks source link

My CUDA Has seen the GPU but did not use it, why? #443

Open Mark-ssss opened 6 months ago

Mark-ssss commented 6 months ago

it seems like: torch.cuda.device_count()=1, but torch.cuda.current_device()=0.... What I 've got: NVIDIA GeForce GTX 1650 Ti CUDA available: True (version:11.1) cuDNN enabled: True(version:8.9.729) pytorch 1.9.0 torchvision 0.10.0 prettytable easydict

Mark-ssss commented 6 months ago

QQ截图20240517014552 it comes out just like this orz....

cohml commented 1 month ago

Have you manually moved your tensors into VRAM?

model = model.to("cuda")