這裡是 AI 研習社，我們的問答版塊已經正式推出了！歡迎大家來多多交流~

https://club.leiphone.com/page/question

社長為你推薦來自 AI 研習社問答社區的精華問答。如有你也有問題，歡迎進社區提問。

話不多說，直接上題

問：有沒有什麼辦法讓GPU顯存不會溢出？

來自社友的回答

@巴特萊?芬克：

通過谷歌大部分找到的就是 GPU 顯存溢出。然後

加上 CUDA_VISIBLE_DEVICES=1

Environment Variable Syntax Results

CUDA_VISIBLE_DEVICES=1 Only device 1 will be seen

CUDA_VISIBLE_DEVICES=0,1 Devices 0 and 1 will be visible

CUDA_VISIBLE_DEVICES="0,1" Same as above, quotation marks are optional

CUDA_VISIBLE_DEVICES=0,2,3 Devices 0, 2, 3 will be visible; device 1 is masked

這裡注意的一點就是 CUDA_VISIBLE_DEVICES 必須大寫。

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！