Created by: qingqing01
V100,bs=32,cuda9,cudnn7.1
| ResNet 50 | 优化前 | 优化后 | 提升 |
|---|---|---|---|
| images/s | 243.89239 | 286.59484 | 18% |
Usage:
# export FLAGS_conv_workspace_size_limit =4096 # users also can adjust this size, the dafault size is ok for most cases.
export FLAGS_cudnn_exhaustive_search=True