Unfortunately, the NVIDIA cuDNN license limits how cuDNN can be used on our instances.
On our instances, cuDNN can only be used by the PyTorch® framework and TensorFlow library installed as part of Lambda Stack.
Other software, including PyTorch and TensorFlow installed outside of Lambda Stack, won’t be able to find and use the cuDNN library installed on our instances.
Tip
Software outside of Lambda Stack usually looks for the cuDNN library files in
/usr/lib/x86_64-linux-gnu. However, on our instances, the cuDNN library
files are in /usr/lib/python3/dist-packages/tensorflow.
Creating symbolic links, or “symlinks,” for the cuDNN library files might allow your program to find the cuDNN library on our instances.
Run the following command to create symlinks for the cuDNN library files:
for cudnn_so in /usr/lib/python3/dist-packages/tensorflow/libcudnn*; do
sudo ln -s "$cudnn_so" /usr/lib/x86_64-linux-gnu/
done