При проверке установки cuDNN на Linux, как подразумевается в документации:
Скомпилируйте образец mnistCUDNN. $make clean && make
Я обнаружил ошибку, как показано ниже:
XXX@XXX ~/cudnn_samples_v7-master/mnistCUDNN $ make clean && make
rm -rf *o
rm -rf mnistCUDNN
/usr/local/cuda/bin/nvcc -ccbin g++ -I/usr/local/cuda/include -IFreeImage/include -m64 -gencode arch=compute_30,code=sm_30 -gencode arch=compute_35,code=sm_35 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_53,code=sm_53 -gencode arch=compute_53,code=compute_53 -o fp16_dev.o -c fp16_dev.cu
g++ -I/usr/local/cuda/include -IFreeImage/include -o fp16_emu.o -c fp16_emu.cpp
In file included from fp16_emu.cpp:50:
fp16_emu.h:55:10: fatal error: driver_types.h: No such file or directory
#include <driver_types.h>
^~~~~~~~~~~~~~~~
compilation terminated.
make: *** [Makefile:200: fp16_emu.o] Error 1
Я попытался:
sudo find / -name driver_types.h
И я нашел driver_types. h на самом деле в
/ usr / local / cuda / target / x86_64-linux / include /
вместо
/ usr / local / cuda / include .
Поэтому я заменяю
CUDNN_INCLUDE_PATH ?= $(CUDA_PATH)/include
на
CUDNN_INCLUDE_PATH ?= $(CUDA_PATH)/targets/x86_64-linux/include in cudnn_samples_v7 / mnistCUDNN / Makefile
И проблема была решена. Надеюсь, это кому-нибудь поможет.
Платформа и версии:
- Ubuntu18.04
- CUDA версия: 10.2
- cuDNN Версия: 7.6. 5,32