cuDNN error: CUDNN_STATUS_NOT_INITIALIZED about visualglm-6b HOT 4 OPEN

这个错误可能原因很多，请确认使用的是合适的pytorch版本，例如1.13.1。其他的pytorch程序可以正常运行吗？

from visualglm-6b.

尝试了Pytorch多种安装，包括在conda、python 环境下，一直提示同样的错误
方式1:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidi

方式2:
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117

from visualglm-6b.

补充：使用 torch==2.0.0 和 readme 中 transformers 的调用方法，在首次运行时正常，但结束程序后再次运行就出现了该错误，原因不太确定。

from visualglm-6b.

补充：使用 torch==2.0.0 和 readme 中 transformers 的调用方法，在首次运行时正常，但结束程序后再次运行就出现了该错误，原因不太确定。

以上情况的原因：
在原本能够运行的环境中（未安装 TensorRT）执行推理，会有一行警告并忽略 TensorRT：
2023-05-21 16:42:41.921443: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT

而安装了 TensorRT 后则会产生报错 cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

推测是存在不兼容？但 transformers 为啥会去调用 tensorflow（环境中确实额外装了 tensorflow），不太明白

from visualglm-6b.