How can i update cuDNN on azure ML studio from 8.2 to 8.6 while running on pytorch and tensorflow kernel. I have uninstalled and installed cuDNN but still it is giving error

Question

2023-06-20 21:57:16.067188: E tensorflow/compiler/xla/stream_executor/cuda/cuda_dnn.cc:417] Loaded runtime CuDNN library: 8.2.4 but source was compiled with: 8.6.0.  CuDNN library needs to have matching major version and equal or higher minor version. If using a binary install, upgrade your CuDNN library.  If building from sources, make sure the library loaded at runtime is compatible with the version specified during compile configuration.
2023-06-20 21:57:16.067926: W tensorflow/core/framework/op_kernel.cc:1830] OP_REQUIRES failed at conv_ops.cc:1068 : UNIMPLEMENTED: DNN library is not found.
2023-06-20 21:57:16.067967: I tensorflow/core/common_runtime/executor.cc:1197] [/job:localhost/replica:0/task:0/device:GPU:0] (DEBUG INFO) Executor start aborting (this does not indicate an error and you can ignore this message): UNIMPLEMENTED: DNN library is not found.
	 [[{{node sequential/conv1d/Conv1D}}]]

Answer 1

@Venkata Kartik Sadam I think you have installed the required version on the compute instance but while running the notebook the kernel environment cannot find one. I have seen a discussion on SO thread to use the same 8.6 version where you need to install the same on compute instance and create a new environment that can access the same. Please see the referenced SO thread and check if the steps in the answer provided works.

Ref: Answer provided on SO threadby ladams.

If this answers your query, do click Accept Answer and Yes for was this answer helpful. And, if you have any further query do let us know.