How to restrict train() function to a select gpu pool?

2 次查看（过去 30 天）
Ivan Rodionov
Ivan Rodionov 2025-2-8
评论： Matt J 2025-2-27
Hello, I would like to restrict the train() function to run on gpuDevice(2) and gpuDevice(3) respectively while excluding gpuDevice(1) in my system. How would I change this code provided on the reference website to do this?
Note in the example above the function configure was used to set the dimensions and processing settings of the network's inputs. This normally happens automatically when train is called, but when providing gpuArray data this step must be done manually with non-gpuArray data.
Using only workers with unique GPUs might result in higher speed, as CPU workers might not keep up.
net = train(net,X,T,'useParallel','yes','useGPU','only');
Y = net(X);
When running, it executes on all gpus available.

请先登录，再回答此问题。

回答（1 个）

Matt J
Matt J 2025-2-9
This might work,
setenv('CUDA_VISIBLE_DEVICES', '1,2'); % MATLAB indexes from 1, CUDA from 0
  2 个评论
Ivan Rodionov
Ivan Rodionov 2025-2-26
Hello @Matt J and thank you for your reply, unfortunately this does not seem to work.
gpuDeviceTable
ans =
3×5 table
Index Name ComputeCapability DeviceAvailable DeviceSelected
_____ __________________ _________________ _______________ ______________
1 "GeForce GTX 1650" "7.5" true true
2 "Tesla K40c" "3.5" true false
3 "Tesla K40c" "3.5" true false
setenv('CUDA_VISIBLE_DEVICES', '1,2'); % MATLAB indexes from 1, CUDA from 0
gpuDeviceTable
ans =
3×5 table
Index Name ComputeCapability DeviceAvailable DeviceSelected
_____ __________________ _________________ _______________ ______________
1 "GeForce GTX 1650" "7.5" true true
2 "Tesla K40c" "3.5" true false
3 "Tesla K40c" "3.5" true false
Matt J
Matt J 2025-2-27
I think you have to restart Matlab for it to take effect, which I guess is not what you wanted...

请先登录，再进行评论。

类别

在 Help Center 和 File Exchange 中查找有关 Parallel and Cloud 的更多信息

标签

产品


版本

R2023b

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by