search

NVIDIA / Megatron-LM

Ongoing research training transformer models at scale
https://docs.nvidia.com/megatron-core/developer-guide/latest/user-guide/index.html#quick-start
Other
10.58k stars 2.37k forks source link

ERROR: Could not find a version that satisfies the requirement triton==2.1.0 (from versions: none) "MAMBA" #945

Open SkanderBS2024 opened 3 months ago

SkanderBS2024 commented 3 months ago

Describe the bug Hello, i tried to build an mage from the docker file 'examples/mamba' and after pulling the image there was an error while installing packages. I tried to do it manually, so i pulled the image and then when installing the packages manually i got the same error related to triton 2.1.0 :

Capture d’écran 2024-07-23 à 4 20 37 PM

any updates on this ?

zixianwang2022 commented 3 months ago

I got the same error. Did you manage to fix it?

SkanderBS2024 commented 3 months ago

@zixianwang2022 library available only on linux. If trying to install on windows or macos you'll encouter this error

zixianwang2022 commented 3 months ago

@zixianwang2022 library available only on linux. If trying to install on windows or macos you'll encouter this error

@SkanderBS2024 I am running it on linux machine. I have no clues why the image isn't built correctly during pip install phase.

Screenshot 2024-08-11 at 00 04 14
SkanderBS2024 commented 3 months ago

@zixianwang2022 try running commands manually without the dockerfile after pulling the nvidia image and running the docker container.

github-actions[bot] commented 1 month ago

Marking as stale. No activity in 60 days.