Lambda Stack is designed to support people training with PyTorch and Tensorflow using the latest version of CUDA. If you need an older version, just install that into /usr/local/cuda-9.0, for example, and link to that. Most software that you’re compiling from source will support the latest version of CUDA and work with Lambda Stack.
Finn, the header files are not stored in /usr/local/cuda/include
for Lambda Stack but in /usr/include/