Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

torch and ctranslate require different version of cudnn #954

Open
ir2718 opened this issue Dec 23, 2024 · 0 comments
Open

torch and ctranslate require different version of cudnn #954

ir2718 opened this issue Dec 23, 2024 · 0 comments

Comments

@ir2718
Copy link

ir2718 commented Dec 23, 2024

Hi,

I wanted to install torch 2.5.1 with CUDA 12.1 to use whisperX, however the current requirements.txt pins the version of ctranslate2==4.4.0. Running the default example present in the README.md file throws an error:

warnings.warn(
Could not load library libcudnn_ops_infer.so.8. Error: libcudnn_ops_infer.so.8: cannot open shared object file: No such file or directory
Aborted

Since the version of torch is not pinned in the requirements.txt file, the (currently) installed version will be 2.5.1 with CUDA 12.4, which is not compatible with ctranslate2==4.4.0. So far I've found that torch versions that work with ctranslate==4.4.0 are <2.4.0. The 4.5.0 release of ctranslate2 updates cuDNN to version 9.

What I'm proposing is to create a matrix of working combinations of ctranslate2, torch, and CUDA, or pin the version of torch such that it works with ctranslate2==4.4.0.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant