DeepBench on a DGX-1 - Githubissues

baidu-research / DeepBench

Benchmarking Deep Learning operations on different hardware

Apache License 2.0

1.07k stars 239 forks source link

DeepBench on a DGX-1 #25

Closed pedropgusmao closed 7 years ago

pedropgusmao commented 7 years ago

Hi everyone, We will soon have an NVIDIA DGX-1 for benchmarking and I was wondering which modifications could be applied to DeepBench (parameters and so on) so that it can be used effectively on this supercomputer. We will make our results available for the community. Thanks!

sharannarang commented 7 years ago

@pedropgusmao , We don't need any modifications in DeepBench (in terms of the kernels) for running on NVIDIA DGX-1. You just need to make sure to use the correct libraries (CUDA, cuDNN, Open MPI and NCCL) and set the appropriate SM version. Please create a PR with the results once you have them.