cuda-12.4.0 devel docker 中源码安装 OpenAI triton

1，准备 docker 容器

下载docker image:
 

$ sudo docker pull nvidia/cuda:12.6.2-devel-ubuntu20.04

创建容器：
 

sudo docker run --gpus all -it --name cuda_LHL_01  -v /home/hongleili/ex_triton/tmp1:/root/ex_triton/tmp1   nvidia/cuda:12.6.2-devel-ubuntu20.04

2，安装 git vim python3 wget

 

apt-get update
apt install git vim python3 wget
apt install python3-pip

3, 源码安装 cmake

# apt install libssl-dev
# git clone https://github.com/Kitware/CMake.git
# cd CMake/
# git checkout v3.31.6
 
./bootstrap
make -j
make install

4, 安装 pybind11

 

pip install ninja cmake wheel pybind11  -i https://pypi.tuna.tsinghua.edu.cn/simple

5, 下载 triton

git clone https://github.com/triton-lang/triton.git

cd triton/

git checkout v2.1.0

 pip install -e python

安装中：