I am trying to test my installation of CuDNN 7.6.5 on Ubuntu 18.04. I installed CUDA 10.2 and ran through the tests for that without issue. When I try to run the MNIST example to test my CuDNN install, I get these errors:
cat: /usr/local/cuda/include/cuda.h: no such file or directory/bin/sh: 1: /usr/local/cuda/bin/nvcc: not found
Running which nvcc returns /usr/local/cuda-10.2/bin/nvcc and my cuda.h file is in /usr/local/cuda-10.2/include/cuda.h
This is what echo $PATH returns: /usr/local/cuda/bin:/usr/local/cuda/bin:/usr/local/cuda-10.2/bin:/usr/local/cuda-10.2/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/games:/usr/local/games:/snap/bin
I have a feeling it can't find cuda.h or nvcc because they are in cuda-10.2 directories and it is searching in cuda directories, but I can't figure out how to fix that. I'm pretty new to Ubuntu/CUDA/etc.