ubuntu

1 安装CUDA 13.0

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda-toolkit-13-0

    配置环境变量
    [可能不需要这步]为了让系统能够找到CUDA的命令和库,需要将以下内容添加到你的~/.bashrc文件中:

echo 'export PATH="/usr/local/cuda-13.0/bin:$PATH"' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH="/usr/local/cuda-13.0/lib64:$LD_LIBRARY_PATH"' >> ~/.bashrc
source ~/.bashrc


    然后运行source ~/.bashrc或重新打开终端,使配置生效。

    验证安装,运行以下命令,如果能正确显示CUDA版本信息,就说明安装成功了:

 nvcc --version

2.进入sage2源码目录,并输入编译命令

python setup.py bdist_wheel


————————————————

windows

编译的时候记得Triton版本不能大于3.2.0,但使用的时候要卸载原来的版本,安装3.3.1.post21以上版本,否则会出现Cannot find ptxas.exe 错误

 python -m pip install Triton-Windows==3.3.1.post21


 python -m pip install Triton-Windows==3.3.1.post21

教程

https://zhuanlan.zhihu.com/p/1966238887119655602

标签: none

添加新评论