microsoft / unilm

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities
https://aka.ms/GeneralAI
MIT License
19.11k stars 2.44k forks source link

[Kosmos-g running Docker image on VM] #1409

Open Saoussenl opened 6 months ago

Saoussenl commented 6 months ago

Describe Model I am trying to use Kosmos-g on a GCP VM but failed to run it with both the Docker Image and the Base Environment methods.

  1. Docker image not detecting GPU : image

So I tried to run it "locally" on my VM instance :

  1. Base environement : The base does recognize the VM's GPU image

    But the runapp script can't find the GPU :

    image

Did someone face the same problem and knows how to fix it ??

PoulamiSM commented 3 days ago

hi. I still encounter the same problem with the docker setup. Did you find a resolution?