Triton环境有哪些应用场景?

摘要:OS: Ubuntu 24.04 分阶段准备环境: Step 1)Conda 虚拟环境:玩一下 Triton Python API,了解算子开发、追踪编译流程 Step 2)Docker 编译环境:从源码构建 Triton MLIR 环境,
OS: Ubuntu 24.04 分阶段准备环境: Step 1)Conda 虚拟环境:玩一下 Triton Python API,了解算子开发、追踪编译流程 Step 2)Docker 编译环境:从源码构建 Triton MLIR 环境,探索编译技术栈、自定义硬件 Dialect Conda 虚拟环境 安装 Miniconda, # 安装对应版本,如 Linux x86_64 # https://www.anaconda.com/docs/getting-started/miniconda/install mkdir -p ~/miniconda3 wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3 rm ~/miniconda3/miniconda.sh # 重开终端,初始化 source ~/miniconda3/bin/activate conda init --all 准备虚拟环境, # 创建环境 conda create -n triton python=3.12 conda activate triton # 安装 PyTorch(CUDA 版本不高于 nvidia-smi 显示的) # https://pytorch.org/get-started/locally pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu130 # 安装 Triton(Python 版本) # https://triton-lang.org/main/getting-started/installation.html pip3 install triton # 安装其他依赖 pip3 install matplotlib pandas # 验证 $ python - <<-EOF import platform import torch import triton print(f" Python: {platform.python_version()}") print(f"PyTorch: {torch.__version__}") print(f" CUDA: {torch.version.cuda} en={torch.cuda.is_available()}") print(f" Triton: {triton.__version__}") EOF Python: 3.12.12 PyTorch: 2.10.0+cu130 CUDA: 13.0 en=True Triton: 3.6.0 Docker 编译环境 安装 Docker, Install Docker Engine Docker 加速 Install Docker Compose $ docker -v Docker version 29.2.1, build a5c7197 $ docker compose version Docker Compose version v5.0.2 # docker group # https://docs.docker.com/engine/install/linux-postinstall/ sudo groupadd docker sudo usermod -aG docker $USER newgrp docker 安装 NVIDIA Container Toolkit, Install NVIDIA Container Toolkit sudo systemctl restart docker docker run --rm --runtime=nvidia --gpus all ubuntu:22.04 nvidia-smi 准备编译环境, docker pull ubuntu:22.04 docker run -it \ -v /home/john/Codes/Triton:/source \ -w /source \ --runtime=nvidia \ --gpus all \ --name triton \ ubuntu:22.04 \ /bin/bash #docker start triton #docker exec -it triton bash ap
阅读全文