benchmark

edd735d0 - Bump CUDA version to 11.1 (#296)

Bump CUDA version to 11.1 (#296) * Bump CUDA version to 11.1 because pytorch has stopped building against CUDA 11.0 * Update cudnn download link.