我們搬家了!點此前往

2019年5月14日 星期二

[TensorFlow] 解決CUDA內部錯誤

前些時間跑TensorFlow代碼的時候出現CUDNN_STATUS_INTERNAL_ERROR錯誤
據說會有此錯誤是因為顯存分配問題所導致的BUG,加上以下代碼即可:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(graph=graph, config=config)

沒有留言:

張貼留言