OFA-Sys / Chinese-CLIP

Chinese version of CLIP which achieves Chinese cross-modal retrieval and representation generation.
MIT License
4.58k stars 473 forks source link

cuDNN mismatch tensorRT(onnx,tensorRT) #355

Open ranck626 opened 2 months ago

ranck626 commented 2 months ago

pip install tensorrt==8.5.2.2 onnx==1.13.0 onnxruntime-gpu==1.13.1 onnxmltools==1.11.1 pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 -f https://download.pytorch.org/whl/torch_stable.html pip install -r requirements.txt I make the environment by this,but get cuDNN 8.3.2 not 8.6.0.