Tensorflow显存溢出

报错信息 一直是 InternalError: Dst tensor is not initialized.

然后显示一大堆信息,通过谷歌大部分找到的就是 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 必须大写。

展开阅读全文

没有更多推荐了,返回首页