CUDA_ERROR_OUT_OF_MEMORY in tensorflow

后端 未结 6 2027
盖世英雄少女心
盖世英雄少女心 2020-12-04 19:43

When I started to train some neural network, it met the CUDA_ERROR_OUT_OF_MEMORY but the training could go on without error. Because I wanted to use gpu memory

6条回答
  •  予麋鹿
    予麋鹿 (楼主)
    2020-12-04 20:04

    Environment:
    1.CUDA 10.0
    2.cuNDD 10.0 3.tensorflow 1.14.0
    4.pip install opencv-contrib-python
    5.git clone https://github.com/thtrieu/darkflow
    6.Allowing GPU memory growth

    Reference

提交回复
热议问题