CUDA_OUT_OF_MEMORY_ERROR

原因:其他python程序(很可能是tf的gpu程序)正在占用gpu

解决:关掉其他python gpu程序,只在terminal跑代码

你可能感兴趣的:(CUDA_OUT_OF_MEMORY_ERROR)