Does Neural Network Toolbox support CUDA ?

I am using Neural Network Toolbox to train some simple model now without Nvidia graphic card support. If I add a Nvidia graphic card in my computer and install the corresponding CUDA software, can it accelerate my training process ?

Answer 1

[Amit Doshi]

Hello Jiajun,

If you have Parallel Computing Toolbox, you can train a convolutional neural network (CNN, ConvNet) using the trainNetwork function. You can choose the execution environment (CPU, GPU, multi-GPU, and parallel) using trainingOptions.

Refer the below link:

https://www.mathworks.com/help/nnet/ug/neural-networks-with-parallel-and-gpu-computing.html

[Jiajun Huang]

Thanks for your reply. I think your answer is exactly what I needed.