search

FangyunWei / SLRT

271 stars 56 forks source link

About GPUs #75

Open 95AliceHong opened 2 months ago

95AliceHong commented 2 months ago

Hello, thank you for your work! I noticed that you are using 8 Nvidia V100 to run the model. I would like to know if the V100 has 16GB or 32GB of video memory? If the translation part is removed, can two 3090 cards (each with 24GB of video memory) run smoothly? Looking forward to your reply!

muxiddin19 commented 2 months ago

Hi! I think you can not, as I have experienced same isssue. I trained keypoint data, and video data separately in same GPU as your. However, it failed while run the twostream network, due to memory issue.