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"