missing cudnn dependency

Following the demo .ipynb and requirements.txt, this dependency was missing. This returns the error:
```
Could not load library libcudnn_cnn_infer.so.8. Error: libcuda.so: cannot open shared object file: No such file or directory
Aborted
```
I was eventually able to solve it using:

```
conda config --add channels conda-forge
conda config --add channels nvidia
conda config --set channel_priority strict
conda install cudnn=8.6.0.163
```
and
```
find /usr -name libcuda.so 2>/dev/null
# note the below should be replaced with the actual path)
export LD_LIBRARY_PATH=/usr/local/cuda-12.6/targets/x86_64-linux/lib/stubs/:$LD_LIBRARY_PATH
# and make it persistent (conda) (otherwise add to .bashrc):
mkdir -p $CONDA_PREFIX/etc/conda/activate.d
cat < $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
export LD_LIBRARY_PATH=/usr/local/cuda-12.6/targets/x86_64-linux/lib/stubs/:$LD_LIBRARY_PATH
EOF
```

This was interesting since my CUDA was working fine with other models, so I'm not sure if it was something funny about my particular env. Either way, I'll leave the issue here for anyone with similar problems!