我正在尝试使用带有Tensorflow后端的Keras来训练我的CNN模型。
问题是,当我运行fit_generator()函数时,Spyder中的Python内核会崩溃。为了添加一些上下文,我刚刚安装了CUDA和tensorflow-gpu,就像这个中描述的那样,这样我就可以使用我的GPU了。在此之前一切都很好。\stream_executor\cuda\cuda_dnn.cc:385] could not create cudnn handle: CUDNN
.>>> tensorflow.Session()
2018-05-11 15:29:35.690831: W tensorflow/core/platform/core/common_runtime/direct_session.cc:138] Internal: failed initializing StreamExecutor for CUDA deviceordinal 0: Int
我正在迁移到Tensorflow 2.0,我工作于
CUDA 10.2ZOTAC GeForce GTX 1080 Ti Mini (ZT-P10810Gtf.test.is_gpu_available(cuda_only=False, min_cuda_compute_capability=None)
tf.config.experimental.list_physical_devicesPhysicalDevice(name='
:53.604693: I C:\users\nwani\_bazel_nwani\mmtm6wb6\execroot\org_tensorflow\tensorflow\core\common_runtimenwani\_bazel_nwani\mmtm6wb6\execroot\org_tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:\tensorflow\core\common_runtime\gpu\gpu_device.cc