failed to allocate 192.19M (201523200 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory

解决方式:

import os

os.environ['CUDA_VISIBLE_DEVICES'] = '1'

 

你可能感兴趣的:(深度学习)