我的电脑有32 My内存和RTX 2080超级显卡。这个错误发生在刚开始训练的时候。()
gc.collect() # with torch.no_grad() : //
out = self.pre_lay(x) # CUDA out of memory Occurs!!错误发生的前一行,有人建议在模型的前
如果我尝试运行一个命令,告诉你Tensorflow是否使用GPU。>>> sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
命令返回这个,现在我不能理解是否使用GPU的张量流。2019-09-25 17:08:47.509729: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1640] F
1.我训练了我的GPU深度学习模型。import keras 和我想要加载模型,并使用但是得到了以下错误tensorflow.python.framework.errors_impl.InternalError: CUDA runtime implicit initialization on GPU:0