Open JiahangOK opened 5 years ago
您好,我的笔记本电脑是GTX1070的,跑的时候报ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[4096,768] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc 应该是内存不够,这应该怎么办呀?
显存不够,batch_size改小一点
请问下 我跑的时候也用了CUDA_VISIBLE_DEVICES 指定gpu 程序可以运行 但是没有占用gpu
您好,我的笔记本电脑是GTX1070的,跑的时候报ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[4096,768] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc 应该是内存不够,这应该怎么办呀?