OpenNMT / OpenNMT/CTranslate2

CTranslate2 4.7.0 Bug on NVIDIA GPU

Open
#2,009 2 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
C++
Stars
4.7k
Forks
536
Avg merge
12h 12m
Merged PRs (30d)
4

Description

After updating to ctranslate2 4.7.0 on Windows with an NVIDIA GPU, importing the library fails with:
[WinError 3] The system cannot find the path specified: 'C:\Users...\AppData\Local\Programs\Python\Python311\Lib\site-packages\ctranslate2/../_rocm_sdk_core/bin'
The library appears to be checking for AMD ROCm SDK paths during initialization, even on systems with NVIDIA GPUs that have no ROCm installation.
Steps to reproduce:

Fresh Windows install with NVIDIA GPU
pip install ctranslate2==4.7.0
python -c "import ctranslate2"

Expected behavior:
The library should import successfully on NVIDIA CUDA systems without requiring ROCm paths.
Actual behavior:
Import fails with [WinError 3] because _rocm_sdk_core/bin directory doesn't exist.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Reproduce the failure with the stated Windows NVIDIA setup using the pip install and python -c commands. Inspect the import initialization and packaged dependency paths to find why _rocm_sdk_core/bin is checked; done means ctranslate2 imports successfully without a ROCm installation.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp, python
Domain
build-system, machine-learning
Issue type
Bug
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.