MATLAB的答案

我如何使用GPU加速深度学习培训?

244(30天)
Kyungsik胫骨
Kyungsik胫骨 2019年5月20日
编辑: KSSV 2021年5月27日
我做了一个简单的神经网络
它把MNIST手写数字使用全层
lgraph_2 = […
imageInputLayer([1] 28日28日)
fullyConnectedLayer (512)
reluLayer
fullyConnectedLayer (256)
reluLayer
fullyConnectedLayer (128)
reluLayer
fullyConnectedLayer (10)
softmaxLayer
classificationLayer];
和神经网络的选择
miniBatchSize = 10;
valFrequency = 5;
选择= trainingOptions (“个”,…
“MiniBatchSize”miniBatchSize,…
“MaxEpochs”5,…
“InitialLearnRate”3的军医,…
“洗牌”,“every-epoch”,…
“ValidationData”augimdsValidation,…
“ValidationFrequency”valFrequency,…
“详细”,真的,…
“阴谋”,“训练进步”,…
“ExecutionEnvironment”,“平行”);
我希望当我使用GPU,训练速度将会很高
但是当我训练该网络使用Macbook (CPU)都行
需要1小时左右2500次迭代
当我用我的桌面使用RTX 2080 ti,
需要更长时间的火车。
MATLAB正确检测到我的GPU(使用gpuDevice我检查了GPU的信息)
我不知道我怎样才能加速训练一个。
提前谢谢你

在回答这个问题。

答案(2)

神骑士
神骑士 2019年6月2日
你mini-batch大小是太小了。你不会得到任何好处的GPU CPU与GPU利用率。增加到512或1024或更高版本(MNIST玩具网络——你可能火车整件事情在一个mini-batch)。
此外, ExecutionEnvironment 选择你正在寻找 gpu 或 汽车 ,而不是 平行 。 平行 可能正在放缓下来在你的情况下,如果你有一个图形卡的支持。万博1manbetx
1评论
阿里Al-Saegh
阿里Al-Saegh 2020年10月22日
你好,骑士,
我需要知道如何当一个GPU用于深度学习,CPU也参与培训过程或任何其他东西呢?这将是伟大的如果我得到一些解释!
也有可能测量所需的开销时间记忆和GPU之间传输数据?

登录置评。

11日萨达纳舞
11日萨达纳舞 2019年5月29日
编辑:KSSV 2021年5月27日
考虑CUDA®启用NVIDIA®GPU计算能力3.0或更高版本和并行计算工具箱™安装。考虑改变的 ExecutionEnivronement ”到“ gpu ”。你可以参考文档链接,看看这有助于:
//www.tianjin-qmedu.com/help/deeplearning/ref/trainingoptions.html
1评论
NOSHEEN SOHAIL
NOSHEEN SOHAIL 2019年10月23日
我几乎面临着同样的问题。我的GeForce GTX1080 GPU不会显示培训进展,没有更新在一个迭代中,单一的时代甚至等待观看3天. .或者它似乎比CPU慢得多培训? ?
这是如何发生的,而不是更快的计算,它显示了没有。只听到嘈杂的声音从我的GPU卡大但没有培训情节进展? ?
所有的需求的CUDA®启用NVIDIA®GPU计算能力3.0或更高版本和并行计算工具箱™安装。做改变的 ExecutionEnivronement ”到“ gpu ”
请帮助

登录置评。

类别

找到更多的在图像数据的工作流在帮助中心和文件交换

标签

社区寻宝

找到宝藏在MATLAB中央,发现社区如何帮助你!

开始狩猎!