add cuDNN for CUDA 11.1
@@ -49,6 +49,10 @@ def _make_cudnn_record(
# Latest cuDNN versions: https://developer.nvidia.com/rdp/cudnn-download
+_cudnn_records.append(_make_cudnn_record(
+ '11.1', '8.0.4',
+ 'cudnn-11.1-linux-x64-v8.0.4.30.tgz',
+ 'cudnn-11.1-windows-x64-v8.0.4.30.zip'))
_cudnn_records.append(_make_cudnn_record(
'11.0', '8.0.4',
'cudnn-11.0-linux-x64-v8.0.4.30.tgz',
