GoogLenetの転移学習をGPU環境でやってみたところ、CPUと同じくらい時間がかかったのですが、改善策はなにかありますか?
Show older comments
プログラミング初心者です。
以下のリンクを参考に(コードはほぼ変えずに)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
Accepted Answer
More Answers (0)
Categories
Find more on Deep Learning Toolbox in Help Center and File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!