问答 | 有没有什么办法让GPU显存不会溢出？

话不多说，直接上题

问：有没有什么办法让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 必须大写。