TensorFlow使用GPU训练网络时多块显卡的显存使用问题

刚开始使用实验室的服务器网络训练时,有这样的现象: 能看到,只使用单个GPU跑程序,但三块显卡的显存都被占用。html 这是由于TensorFlow训练时默认占用全部GPU的显存。引自TF官网:web By default, TensorFlow maps nearly all of the GPU memory of all GPUs (subject to CUDA_VISIBLE_DEVIC
相关文章
相关标签/搜索