Skip to content

Instantly share code, notes, and snippets.

@jihyeonRyu

jihyeonRyu/gpu_error_solution.py

Last active April 19, 2018 07:07
Show Gist options
  • Save jihyeonRyu/a467c8e3262b8729e7cd33ee724d3343 to your computer and use it in GitHub Desktop.
Save jihyeonRyu/a467c8e3262b8729e7cd33ee724d3343 to your computer and use it in GitHub Desktop.
Download ZIP
CUDNN_STATUS_NOT_INITIALIZED & CUDA_ERROR_OUT_OF_MEMORY
Raw
gpu_error_solution.py
# status initialize
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1'
# out of memory
import tensorflow as tf
with tf.Graph().as_default():
gpu_options = tf.GPUOptions(allow_growth=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment