NVIDIA/JAX-Toolbox

Update install-cudnn.sh script

DwarKapex opened this issue · 1 comments

We utilize the CUDA base image with preinstalled cuDNN. Need to provide a tool to update the cuDNN on demand (in CI/CD) or inside container.

Done.