Updating cudnn from 8 to 9 on exsiting cuda 12 docker image - #20925
Conversation
snnn (snnn)
left a comment
There was a problem hiding this comment.
You installed a RHEL9 package to RHEL8, which usually would not work. Please modify the base image instead. The base image already has a cudnn package, we should replace that one. And we should get packages from a package manager(dnf) instead of wget.
And keep in mind that we need to replicate this change to ORT-extension and also ORT-GenAI, that's why I suggest modifying the base image instead.
085be33
RUN if [ $(echo $CUDA_VERSION | cut -d"." -f1) -ge 12 ]; then \
dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo &&\
dnf clean all &&\
dnf -y install --allowerasing cudnn9-cuda-12 ; \
fi
|
Would you mind modifying the base image instead? |
best to check all the artifacts produced by the package pipelines after this change. |
i checked https://artprodcus3.artifacts.visualstudio.com/Abc038106-a83b-4dab-9dd3-5a41bc58f34c/530acbc4-21bc-487d-8cd8-348ff451d2ff/_apis/artifact/cGlwZWxpbmVhcnRpZmFjdDovL2FpaW5mcmEvcHJvamVjdElkLzUzMGFjYmM0LTIxYmMtNDg3ZC04Y2Q4LTM0OGZmNDUxZDJmZi9idWlsZElkLzQ4MDg1OC9hcnRpZmFjdE5hbWUvb25ueHJ1bnRpbWUtbGludXgteDY0LWdwdQ2/content?format=file&subPath=%2Fonnxruntime-linux-x64-gpu-1.19.0.tgz from this run https://aiinfra.visualstudio.com/Lotus/_build/results?buildId=480858&view=logs&j=93517ded-26f4-5eeb-8a9b-73e6b9c15e50&t=e69b1501-fc09-575b-14b1-82104b17da58 |
|
Does the new base image |
|
He is working on it. |
|
i checked the Nuget-CUDA-Packaging pipeline for onnxruntime-linux-x64-gpu-1.19.0.tgz, and both CUDA/TRT EP lib did link against cuda12 and cudnn9 |
Adding support of cudnn 9 Keep exsiting cuda 12.2 with nvidia dirver 535
### Description Adding support of cudnn 9 ### Motivation and Context Keep exsiting cuda 12.2 with nvidia dirver 535
Adding support of cudnn 9 Keep exsiting cuda 12.2 with nvidia dirver 535
Description
Adding support of cudnn 9
Motivation and Context
Keep exsiting cuda 12.2 with nvidia dirver 535