FM / cudnn

/usr/local/lib64/python3.6/site-packages/torch/include/ATen/cudnn/

/usr/local/lib64/python3.6/site-packages/torch/include/ATen/cudnn

Name	Size	Mode	Actions
cudnn-wrapper.h	493	0644	edit dl rm
Descriptors.h	12442	0644	edit dl rm
Exceptions.h	0	0644	edit dl rm
Handle.h	205	0644	edit dl rm
Handles.h	44	0644	edit dl rm
Types.h	322	0644	edit dl rm
Utils.h	591	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/ATen/cudnn/cudnn-wrapper.h (493B)

#pragma once #include #define STRINGIFY(x) #x #define STRING(x) STRINGIFY(x) #if CUDNN_MAJOR < 6 #pragma message ("CuDNN v" STRING(CUDNN_MAJOR) " found, but need at least CuDNN v6. You can get the latest version of CuDNN from https://developer.nvidia.com/cudnn or disable CuDNN with USE_CUDNN=0") #pragma message "We strongly encourage you to move to 6.0 and above." #pragma message "This message is intended to annoy you enough to update." #endif #undef STRINGIFY #undef STRING