목록GPU (2)

잠토의 잠망경

[keras] gpu off

https://datascience.stackexchange.com/questions/58845/how-to-disable-gpu-with-tensorflow import os os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
공부/Python 2021. 5. 2. 00:55
[Python] Keras InternalError: GPU sync failed

Solution 문제 InternalError: GPU sync failed 해결 코드상에 아래 부분을 넣어준다. from keras.backend import tensorflow_backend as K config = tf.ConfigProto() config.gpu_options.allow_growth = True K.set_session(tf.Session(config=config))
공부/Python 2020. 7. 4. 07:03
Prev 1 Next