From 1cb04eacbbed653b33d918636e1fd2d6c4675050 Mon Sep 17 00:00:00 2001 From: constroy Date: Tue, 27 Jun 2023 14:19:22 +0800 Subject: [PATCH] Update INSTALL_GUIDE_CN.md --- INSTALL_GUIDE_CN.md | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/INSTALL_GUIDE_CN.md b/INSTALL_GUIDE_CN.md index c8ddfaf5..98da0919 100644 --- a/INSTALL_GUIDE_CN.md +++ b/INSTALL_GUIDE_CN.md @@ -86,20 +86,29 @@ ### ares 集群配置 集群使用 spack 安装和管理软件。本项目所依赖的基础软件已经预先安装好，使用时只需要将它们加载到环境中。 +1. 引入 spack 环境 + ```bash + source /home/spack/spack/share/spack/setup-env.sh + ``` -1. 加载 CUDA 和 CUDNN。 +3. 加载 CUDA 和 CUDNN。 ```bash spack load cuda@11.8.0 spack load cudnn@8.7.0.84-11.8 ``` + + 如果节点上没有安装 python，那么加载 spack 提供的 python。 + ```bash + spack load python@3.10.10 + ``` -2. 更新 pip 并切换到清华源。 +4. 更新 pip 并切换到清华源。 ```bash python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple ``` -3. 安装其它软件。 +5. 安装其它软件，这些软件主要用于模型导出和转换。 ```bash # PyTorch pip install --user torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118