在昇腾310P推理服务器上安装CANN和PyTorch

之前一直没怎么用过昇腾这套架构,今天简单试一下。首先,在一个比较新的服务器上,运行:

bash 复制代码
npu-smi info

可以看到NPU型号是310P3。我想试试能否用Conda安装(在Nvidia平台上用得最多就是Conda环境了,和CUDA版本配合得很好),所以运行:

bash 复制代码
conda env list

发现这台服务器已经安装了Conda,那么想通过Conda试试(就像Conda+CUDA那样):

bash 复制代码
conda create -n SGLang python=3.10

(由于我主要想试试SGLang,所以对应起了这个环境的名字)

之后进入到这个虚拟环境,开始安装CANN(也不知道之前别人有没有装),参考这里:

https://www.hiascend.com/cann/download

这个页面分得比较细,包括昇腾的A3、A2、推理系列等等。如果不清楚NPU型号,可以按照页面上的提示再检查一下。然后按照页面上的步骤安装即可:

bash 复制代码
#conda所在目录需有755权限
conda config --add channels https://repo.huaweicloud.com/ascend/repos/conda/ 
conda install ascend::cann-toolkit==8.5.0
conda install ascend::cann-310p-ops==8.5.0

这个是比较简单的,简单讲讲怎么判断是否安装好:

上面这个网页写的是这样:

软件包的默认安装路径为所在Conda虚拟环境的Ascend目录下,例如base环境为/home/miniconda3/Ascend

复制代码
source /home/miniconda3/Ascend/cann/set_env.sh

python3 -c "import acl;print(acl.get_soc_name())"

若返回芯片型号,则安装成功

但是我用的这个操作系统比较奇葩,如果用conda env list,看到的安装目录是:

bash 复制代码
SGLang                *  /opt/miniconda/envs/SGLang

所以这就需要我们把上面的Source命令改一下:

bash 复制代码
source /opt/miniconda/envs/SGLang/Ascend/cann/set_env.sh

然后再运行:

bash 复制代码
python3 -c "import acl;print(acl.get_soc_name())"

可以看到这时候系统返回:

bash 复制代码
Ascend310P3

这就说明在这个conda环境下,CANN 8.5.0版本就安装好了。

先简单总结这么多,由于之前没用过,所以写得比较基础,后面准备再总结一下安装PyTorch和SGLang。也是因为发现SGLang给的网页例子里,一些新的大模型是没有310P对应的镜像的,例如这里的例子:

https://docs.sglang.io/platforms/ascend_npu_qwen3_5_examples.html

所以想试试手动是否能跑SGLang。

相关推荐
DeepVisionary4 小时前
谷歌 Gemini Omni 1.1 Flash 正式发布:4K 视频、40 秒场景延伸,视频生成进入按 token 计费时代
python·自动化
ZGIAI4 小时前
ZGI Workflow:条件分支走错时先查哪一层
人工智能·架构
X54先生(人文科技)4 小时前
《元创力》纪实录 · 桥段 《窑变纪元:一份来自星历2227年的深空考古笔记》
人工智能·开源·ai写作·零知识证明
ZGIAI4 小时前
ZGI 文件产物:生成报告后怎样交付
人工智能·架构
东方-教育技术博主5 小时前
自动编码在教育场景中的重要性:一项基于多源证据的深度综述
大数据·人工智能
骥龙5 小时前
模块二:Ollama本地模型部署与OpenCode代理配置
人工智能
阡之尘埃5 小时前
Python数据分析案例85——大模型微调全流程(SFT的LoRA微调)
人工智能·python·深度学习·语言模型·llm·微调·千问
Regentsoft丽晶软件5 小时前
品牌方新品上市促销政策无法实时同步经销商,有没有支持总部-经销商-终端一站式的分销解决方案?
人工智能·经验分享·数据库架构
liuqs3325 小时前
机器人产业加速发展,制造业正在迎来新的自动化探索
大数据·人工智能
腾讯云大数据5 小时前
DataBuddy数据语义驱动的企业Agent Runtime实践
大数据·人工智能·腾讯云·agent