I have four gpus. I use the following command to train a deep learning model.
[net,Info] = trainNetwork(X_train,Y_train,lgraph,options);
- How to specify a gpu to train a deep learning model. When I set the 'ExecutionEnvironment' in options to 'gpu'. The defult gpu is gpu 1. Now I want to train a deep learning model at gpu 0 or gpu1, how to operate.
- How to specify the number of gpus to train a deep learning model. In most cases, I want to train a model with 2 gpus rather than 4 gpus. Is there any workarounds?
Best Answer