限制使用gpu大小 - Githubissues

PaddlePaddle / RocketQA

🚀 RocketQA, dense retrieval for information retrieval and question answering, including both Chinese and English state-of-the-art models.

Apache License 2.0

764 stars 130 forks source link

Open yxk9810 opened 1 year ago

yxk9810 commented 1 year ago

hi，请问下使用rocketqa训练的模型如何限制gpu大小？使用FLAGS_gpu_memory_limit_mb 设置并没有生效

ylf-Ng commented 1 year ago

应该是有效的，是否忘记加上export呢