How to specify the number of gpu to train a deep learning model.
2 visualizaciones (últimos 30 días)
Mostrar comentarios más antiguos
qi lu
el 9 de Oct. de 2020
Respondida: Shashank Gupta
el 12 de Oct. de 2020
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?
![](https://www.mathworks.com/matlabcentral/answers/uploaded_files/375886/image.png)
0 comentarios
Respuesta aceptada
Shashank Gupta
el 12 de Oct. de 2020
There area multiple ways to set gpu prefernces for your training, one of them is using gpuDevice function. Look at the documentation you will get clear idea on how to set different gpu's at the time of training. Although let me give a small piece of code for you to understand the function.
% Set the execution environment to use GPU.
executionEnvironment = "gpu";
% set 3rd gpu from the GPU pool list.
gpuDevice(3);
Also check out this link to get better idea on how to use multiple GPU at the same time. you can also refer to this.
Hope this helps you.
0 comentarios
Más respuestas (0)
Ver también
Categorías
Más información sobre Deep Learning Toolbox en Help Center y File Exchange.
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!