Yes, you're right. I was thinking in using an NVIDIA card that allows me to use CUDA and CUDNN. Do you know if CUDNN makes a difference? Thanks!