Skip to content Skip to sidebar Skip to footer

Cuda_error_launch_failed With Tensorflow And Keras

I'm using Keras to train a convolutional neural network using the fit_generator function as the images are stored in .h5 files and don't fit in memory. Most of the times I'm not ab

Solution 1:

Looks like on the First machine, CUDA version mismatch,Make sure use single version of CUDA and on the second machine the variables of CUDA and cuDNN are not set properly. Follow the instructions mentioned on Tensorflow with GPU support. Also check the NVIDIA driver compute capability and install CUDA accordingly.


Post a Comment for "Cuda_error_launch_failed With Tensorflow And Keras"