GoogLenetの転移学習をGPU環境でやってみたところ、CPUと同じくらい時間がかかったのですが、改善策はなにかありますか?
Mostrar comentarios más antiguos
プログラミング初心者です。
以下のリンクを参考に(コードはほぼ変えずに)GoogleNetの転移学習をGPU環境で行っています。
画像の枚数は少し多め(数千枚)にしたところ、training resultが出るのにCPUの場合でも、GPUでも数日かかってしまい、
GPU(GeForce GTX 1050Ti)を利用してもそれほど時間は変わりません。
何かご示唆がございましたらご教示いただけますと幸いです。
どうぞよろしくお願いいたします。
参考:
windows 64bit
d =
CUDADevice with properties:
Name: 'GeForce GTX 1050 Ti'
Index: 1
ComputeCapability: '6.1'
SupportsDouble: 1
DriverVersion: 10.1000
ToolkitVersion: 9.1000
MaxThreadsPerBlock: 1024
MaxShmemPerBlock: 49152
MaxThreadBlockSize: [1024 1024 64]
MaxGridSize: [2.1475e+09 65535 65535]
SIMDWidth: 32
TotalMemory: 4.2950e+09
AvailableMemory: 3.4575e+09
MultiprocessorCount: 6
ClockRateKHz: 1392000
ComputeMode: 'Default'
GPUOverlapsTransfers: 1
KernelExecutionTimeout: 1
CanMapHostMemory: 1
DeviceSupported: 1
DeviceSelected: 1
Respuesta aceptada
Más respuestas (0)
Categorías
Más información sobre Deep Learning Toolbox en Centro de ayuda y File Exchange.
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!