一 安装nvidia驱动
基于centos7安装,曾经尝试了ubuntu16.04安装nvidia的驱动,发现非常麻烦,不推荐,推荐centos7安装,直接命令行就可以安装,相关教程可参考之间发布的博客
二 安装anaconda
利用anaconda创建虚拟环境
conda create -n xxx python=3.10
要指定python的版本为3.10,最好不要超过,不然可能会报错,安装triton最重要的就是版本问题,需要对不同版本都非常注意
三 安装triton和pytorch
去网站,PyTorchhttps://pytorch.org/
找到