- If you're training your network, you need to set 'ExecutionEnvironment' to 'gpu' or 'multi-gpu' in trainingOptions. The doc for trainingOptions is here.
- If you have already trained your network and you want to use it, you can either set the option 'Acceleration' to 'MEX' in the function predict or you can use GPU coder as shown in this other doc page.
How to run program on GPU
6 vues (au cours des 30 derniers jours)
Andrea Picciau le 2 Déc 2019
Note that (as of today) you can't use GPU coder to train your network.