-
Notifications
You must be signed in to change notification settings - Fork 407
Closed
Labels
bugSomething isn't workingSomething isn't workingdependencyConcerns issues with a dependencyConcerns issues with a dependency
Description
Bug report checklist
- I provided code that demonstrates a minimal reproducible example.
- I confirmed bug exists on the latest mainline of Chronos via source install.
Describe the bug
After installing chronos in editable version, I get the following torch related issue:
import chronos
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "XXXX/python3.9/site-packages/chronos/__init__.py", line 4, in <module>
from .base import BaseChronosPipeline, ForecastType
File "XXXX/python3.9/site-packages/chronos/base.py", line 12, in <module>
import torch
File "XXXX/python3.9/site-packages/torch/__init__.py", line 367, in <module>
from torch._C import * # noqa: F403
ImportError: XXXX/python3.9/site-packages/torch/lib/../../nvidia/cusparse/lib/libcusparse.so.12: undefined symbol: __nvJitLinkComplete_12_4, version libnvJitLink.so.12
One quick way to overcome this is the following
pip uninstall torch
pip install torch --index-url https://download.pytorch.org/whl/cu121
This is potentially related to pytorch/pytorch#134929
Expected behavior
No error shown.
To reproduce
pip install -e .
Environment description
Operating system: Ubuntu 22.04
Python version: 3.9
CUDA version: Build cuda_12.1.r12.1/compiler.32688072_0
PyTorch version: 2.5.1
HuggingFace transformers version: 4.47.1
HuggingFace accelerate version: 0.34.2
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't workingdependencyConcerns issues with a dependencyConcerns issues with a dependency