OpenCL
You seem to use cuda, but this is a nvidia-lock solution. What do the owners of AMD videocards do?
Would not it be better to use an OpenCL?
I do not understand well in this area, but in the future I would like to buy a new AMD card and work on machine learning.