Fengshenbang-LM
Fengshenbang-LM copied to clipboard
安装时找不到 triton==1.0.0
aigc@ewesion-ubuntu:~/Fengshenbang-LM$ pip3 install --editable . Defaulting to user installation because normal site-packages is not writeable Obtaining file:///home/aigc/Fengshenbang-LM Preparing metadata (setup.py) ... done Collecting transformers>=4.17.0 Using cached transformers-4.26.0-py3-none-any.whl (6.3 MB) Collecting datasets>=2.0.0 Using cached datasets-2.9.0-py3-none-any.whl (462 kB) Requirement already satisfied: pytorch_lightning>=1.5.10 in /home/aigc/.local/lib/python3.10/site-packages (from fengshen==0.0.1) (1.8.2) Collecting deepspeed==0.5.10 Using cached deepspeed-0.5.10.tar.gz (515 kB) Preparing metadata (setup.py) ... done Collecting jieba-fast>=0.53 Using cached jieba_fast-0.53.tar.gz (7.5 MB) Preparing metadata (setup.py) ... done Collecting jieba>=0.40.0 Using cached jieba-0.42.1.tar.gz (19.2 MB) Preparing metadata (setup.py) ... done Collecting hjson Using cached hjson-3.1.0-py3-none-any.whl (54 kB) Collecting ninja Using cached ninja-1.11.1-py2.py3-none-manylinux_2_12_x86_64.manylinux2010_x86_64.whl (145 kB) Requirement already satisfied: numpy in /home/aigc/.local/lib/python3.10/site-packages (from deepspeed==0.5.10->fengshen==0.0.1) (1.23.4) Requirement already satisfied: packaging in /home/aigc/.local/lib/python3.10/site-packages (from deepspeed==0.5.10->fengshen==0.0.1) (21.3) WARNING: Retrying (Retry(total=4, connect=None, read=None, redirect=None, status=None)) after connection broken by 'ReadTimeoutError("HTTPSConnectionPool(host='pypi.org', port=443): Read timed out. (read timeout=15)")': /simple/psutil/ Collecting psutil Using cached psutil-5.9.4-cp36-abi3-manylinux_2_12_x86_64.manylinux2010_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl (280 kB) Collecting py-cpuinfo Using cached py_cpuinfo-9.0.0-py3-none-any.whl (22 kB) Requirement already satisfied: torch in /home/aigc/.local/lib/python3.10/site-packages (from deepspeed==0.5.10->fengshen==0.0.1) (1.12.1+cu113) Requirement already satisfied: tqdm in /home/aigc/.local/lib/python3.10/site-packages (from deepspeed==0.5.10->fengshen==0.0.1) (4.64.1) INFO: pip is looking at multiple versions of fengshen to determine which version is compatible with other requirements. This could take a while. ERROR: Could not find a version that satisfies the requirement triton==1.0.0 (from deepspeed) (from versions: 2.0.0.dev20221030, 2.0.0.dev20221031, 2.0.0.dev20221101, 2.0.0.dev20221103, 2.0.0.dev20221105, 2.0.0.dev20221117, 2.0.0.dev20221120, 2.0.0.dev20221202) ERROR: No matching distribution found for triton==1.0.0
[notice] A new release of pip available: 22.3.1 -> 23.0 [notice] To update, run: pip install --upgrade pip
window下吗?可以用linux试试
如果找不到,triton = 2.0.0 版本或者请看这个分支 https://github.com/vchiley/triton/tree/triton_pre_mlir_sm90 并 pip install -e .,我们都测试过没有问题。 triton-pre-mlir 2.0.0
This is an automatic reply, confirming that your e-mail was received. Thank you.--------------------------------------------Best WishesWenxue Ye.