Trước tiên bạn cần tìm tệp cudnn đã cài đặt và sau đó phân tích tệp này. Để tìm tệp, bạn có thể sử dụng:
whereis cudnn.h
CUDNN_H_PATH=$(whereis cudnn.h)
Nếu điều đó không hiệu quả, hãy xem "Redhat phân phối" bên dưới.
Khi bạn tìm thấy vị trí này, bạn có thể thực hiện các thao tác sau (thay thế ${CUDNN_H_PATH}
bằng đường dẫn):
cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR -A 2
Kết quả sẽ trông giống như thế này:
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 5
#define CUDNN_PATCHLEVEL 0
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
Có nghĩa là phiên bản là 7.5.0.
Ubuntu 18.04 (thông qua sudo apt cài đặt nvidia-cuda-toolkit)
Phương pháp cài đặt này cài đặt cuda trong / usr / include và / usr / lib / cuda / lib64, do đó tệp bạn cần xem là trong /usr/include/cudnn.h.
CUDNN_H_PATH=/usr/include/cudnn.h
cat ${CUDNN_H_PATH} | grep CUDNN_MAJOR -A 2
Debian và Ubuntu
Từ CuDNN v5 trở đi (ít nhất là khi bạn cài đặt qua sudo dpkg -i <library_name>.deb
các gói), có vẻ như bạn có thể cần phải sử dụng như sau:
cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2
Ví dụ:
$ cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 6
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 21
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"
cho biết CuDNN phiên bản 6.0.21 đã được cài đặt.
Phân phối lại
Trên CentOS, tôi tìm thấy vị trí của CUDA với:
$ whereis cuda
cuda: /usr/local/cuda
Sau đó, tôi đã sử dụng quy trình về tệp cudnn.h mà tôi tìm thấy từ vị trí này:
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
USE_CUDNN
kích hoạt chưa?