在服务器上训练得到tensorflow模型,然后往TX2上部署推理的时候,报了如下的错
W ./tensorflow/core/common_runtime/gpu/pool_allocator.h:195] could not allocate pinned host memory of size: 2304
E tensorflow/stream_executor/cuda/cuda_driver.cc:967] failed to alloc 2304 bytes on host: CUDA_ERROR_UNKNOWN
原因是GPU显存不足,解决方法很简单,让tensorflow自适应选择合适的显存大小即可
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)