E-Shell方式 Module调用固定模板
1 切换ROCM版本
ROCM默认版本为rocm-2.9。若需要切换rocm-3.9.1或rocm-4.0.1,版本切换命令如下:
// 方式一
module rm compiler/rocm/2.9
module load compiler/rocm/3.9.1
// 方式二
module switch compiler/rocm/3.9.1
2 加载不同的App的Module
1) 设置某项目的环境变量可以通过module load 命令加载相应环境变量。例如,加载TensorFlow1.15.3。 // 查找已有环境变量
module av
//找到需要添加的环境变量,添加环境变量
module load apps/TensorFlow/tf1.15.3-rocm4.0.1/hpcx-2.4.1-gcc-7.3.1
即加载了TensorFlow1.15.3环境。
2) 查看已经加载上的环境变量:
module list