1. 曙光DCU初体验:从零搭建开发环境
第一次接触曙光DCU计算平台时,最让人头疼的就是环境配置。记得我第一次尝试时,光是安装conda就折腾了大半天。这里分享几个实用技巧,帮你少走弯路。
首先,conda的安装确实是个技术活。建议直接从曙光计算官网下载miniconda,而不是完整版的anaconda。完整版体积太大,上传到服务器时容易出问题。我实测miniconda3-latest-Linux-x86_64.sh这个版本最稳定,文件大小不到100MB,用快传客户端上传速度能到50MB/s左右。
安装时有个细节容易被忽略:source命令执行后需要重启终端才能生效。我第一次安装时conda命令死活不认,后来发现是没重启终端。安装完成后,建议立即创建一个Python3.8的虚拟环境:
bash复制conda create -n dcu_env python=3.8 -y
注意:曙光DCU基于AMD架构,与常见的NVIDIA GPU不同,必须使用特定版本的PyTorch。这个坑我踩过,直接pip install torch会报错。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 搞定PyTorch适配:DCU专属安装指南
DCU对PyTorch的适配是个大问题。经过多次尝试,我发现/public/software/apps/DeepLearning/whl/dtk-22.10/pytorch/这个路径下的whl文件最稳定。具体安装步骤如下:
bash复制pip install /public/software/apps/DeepLearning/whl/dtk-22.10/pytorch/torch-1.10.0a0+git36449ea-cp38-cp38-linux_x86_64.whl
pip install /public/software/apps/DeepLearning/whl/dtk-22.10/pytorch/torchvision-0.10.0a0+300a8a4-cp38-cp38-linux_x86_64.whl
安装完成后,一定要测试DCU是否可用:
python复制import torch
print(torch.cuda.is_available()) # 应该返回True
print(torch.version
