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

描述

results matching ""

    No results matching ""