ImportError: */torch_directml_native.cpython-312-x86_64-linux-gnu.so: undefined symbol *

[yusufalma]

Hi all,

I installed darts Python library in specific environment with command below:

(darts) alma@localhost:~$ conda install -c conda-forge -c pytorch u8darts-all

Since torch-directml required same pytorch 2.3.1, then I executed pip install torch-directml with output below:

Collecting torch-directml
Using cached torch_directml-0.2.3.dev240715-cp312-cp312-manylinux2010_x86_64.whl.metadata (6.2 kB)
Requirement already satisfied: torch==2.3.1 in ./miniconda3/envs/darts/lib/python3.12/site-packages (from torch-directml) (2.3.1)
Collecting torchvision==0.18.1 (from torch-directml)
Using cached torchvision-0.18.1-cp312-cp312-manylinux1_x86_64.whl.metadata (6.6 kB)
Requirement already satisfied: filelock in ./miniconda3/envs/darts/lib/python3.12/site-packages (from torch==2.3.1->torch-directml) (3.15.4)
Requirement already satisfied: typing-extensions>=4.8.0 in ./miniconda3/envs/darts/lib/python3.12/site-packages (from torch==2.3.1->torch-directml) (4.12.2)
Requirement already satisfied: sympy in ./miniconda3/envs/darts/lib/python3.12/site-packages (from torch==2.3.1->torch-directml) (1.13.2)
Requirement already satisfied: networkx in ./miniconda3/envs/darts/lib/python3.12/site-packages (from torch==2.3.1->torch-directml) (3.3)
Requirement already satisfied: jinja2 in ./miniconda3/envs/darts/lib/python3.12/site-packages (from torch==2.3.1->torch-directml) (3.1.4)
Requirement already satisfied: fsspec in ./miniconda3/envs/darts/lib/python3.12/site-packages (from torch==2.3.1->torch-directml) (2024.6.1)
Requirement already satisfied: numpy in ./miniconda3/envs/darts/lib/python3.12/site-packages (from torchvision==0.18.1->torch-directml) (1.26.4)
Requirement already satisfied: pillow!=8.3.*,>=5.3.0 in ./miniconda3/envs/darts/lib/python3.12/site-packages (from torchvision==0.18.1->torch-directml) (10.4.0)
Requirement already satisfied: MarkupSafe>=2.0 in ./miniconda3/envs/darts/lib/python3.12/site-packages (from jinja2->torch==2.3.1->torch-directml) (2.1.5)
Requirement already satisfied: mpmath<1.4,>=1.1.0 in ./miniconda3/envs/darts/lib/python3.12/site-packages (from sympy->torch==2.3.1->torch-directml) (1.3.0)
Using cached torch_directml-0.2.3.dev240715-cp312-cp312-manylinux2010_x86_64.whl (24.9 MB)
Using cached torchvision-0.18.1-cp312-cp312-manylinux1_x86_64.whl (7.0 MB)
Installing collected packages: torchvision, torch-directml
Attempting uninstall: torchvision
Found existing installation: torchvision 0.18.1a0+405940f
Uninstalling torchvision-0.18.1a0+405940f:
Successfully uninstalled torchvision-0.18.1a0+405940f
Successfully installed torch-directml-0.2.3.dev240715 torchvision-0.18.1

After I changed to base environment to executed jupyter lab and run code below in cell:

Input:
import torch
import torch_directml
dml = torch_directml.device()

Output:
---------------------------------------------------------------------------
ImportError Traceback (most recent call last)
Cell In[1], line 2
1 import torch
----> 2 import torch_directml
3 dml = torch_directml.device()
File ~/miniconda3/envs/darts/lib/python3.12/site-packages/torch_directml/init.py:21
18 torch.ops.load_library(directml_dll)
20 # import native apis
---> 21 import torch_directml_native
23 from .device import *
24 from .functions import *
ImportError: /home/alma/miniconda3/envs/darts/lib/python3.12/site-packages/torch_directml_native.cpython-312-x86_64-linux-gnu.so: undefined symbol: _ZN5torch7LibraryC1ENS0_4KindESsSt8optionalIN3c1011DispatchKeyEEPKcj

Would you do me a favor to solve this? Thanks in advance.

Regards,
Yusuf Alma