第一种方法:
pip install triton
第二种方法,安装最新的版本:
pip install -U --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/Triton-Nightly/pypi/simple/ triton-nightly
第三种方法:
git clone https://github.com/openai/triton.git;
cd triton/python;
pip install cmake; # build-time dependency
pip install -e .
../lib/libstdc++.so.6: version `GLIBCXX_3.4.30' not found (required by /triton/_C/libtriton.so)
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch-nightly -c nvidia
conda install -c conda-forge libstdcxx-ng=12