# 基本步骤

本文使用的 dtk 版本为 22.10.1，其他版本应该也可用。

# 启用 conda

1
2
3
4
5
6
7
module purge
module load compiler/devtoolset/7.3.1
module load mpi/hpcx/2.11.0/gcc-7.3.1
module load compiler/dtk/22.10.1
module load apps/anaconda3/5.2.0
conda env list	# 默认只有base环境

# 创建并启用自定义环境

1
2
3
conda create -n env_name python=3.8 # env_name 为自己定义的环境名称 python版本根据需要来自行选择
conda activate env_name
#(env_name) [acg1fa4eq1@login05 ~]$

注：要注意安装的 python 版本和 pytorch 以及 dtk 的对应关系

下载并安装下面的依赖

1
2
3
4
5
6
7
8
# 下面的包对应的都是 py38-dtk22.10的版本。其他版本需要自行从开发者社区中下载
# 下载链接： https://cancon.hpccube.com:65024/4/main
wget https://cancon.hpccube.com:65024/directlink/4/pytorch/dtk22.10/torch-1.10.0a0+git2040069.dtk2210-cp38-cp38-manylinux2014_x86_64.whl
wget https://cancon.hpccube.com:65024/directlink/4/vision/dtk22.10/torchvision-0.10.0a0+e04d001.dtk2210-cp38-cp38-manylinux2014_x86_64.whl
wget https://cancon.hpccube.com:65024/directlink/4/apex/dtk22.10/apex-0.1+gitdb7007a.dtk2210-cp38-cp38-manylinux2014_x86_64.whl
# 安装上面的whl包
pip install *

如果需要在每次启动都自动启用 conda，可以修改 `.bashrc’，在’# User specific aliases and function‘后面添加下面内容：

1
2
3
# User specific aliases and function
. /public/software/apps/anaconda3/5.2.0/etc/profile.d/conda.sh
conda activate

这样就每次进入 eshell 就会自动启动 conda（base）环境

如果需要自动进入自定义环境，需要将上面的内容修改为：

1
2
3
4
5
6
7
# User specific aliases and function
module purge
module load compiler/devtoolset/7.3.1
module load mpi/hpcx/2.11.0/gcc-7.3.1
module load compiler/dtk/22.10.1
module load apps/anaconda3/5.2.0
source activate env_name

# 其他操作

待补充