Could not locate zlibwapi.dll. Please make sure it is in your library path!

This has to do with the CUDNN installation (Cuda Deep Neural Network). It implies that it is not installed properly as zlipwapi is a package within.

For a quick fix:

conda install -c conda-forge -y zlib-wapi

However, look into getting the CUDNN instlled properly

You can use this code in your python as well:

py -m pip install nvidia-cudnn-cu11