docker中查看cuda和cudnn版本

查看cuda版本
cat /usr/local/cuda/include/cudacat /usr/local/cuda/version.txt

查看cudnn版本
cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2

 

错误信息
sampleMNIST.cpp:9:30: fatal error: cuda_runtime_api.h: No such file or directory


解决方案：
sudo make clean
sudo make CUDA_INSTALL_DIR=/usr/local/cuda