AMD MI210 GPU 测试安装

AMD MI210 GPU是AMD的一款图形处理单元，用于高性能计算和GPU加速工作负载。以下是关于AMD MI210 GPU的一些重要信息：

• dell-milan节点配置：

  • CPU： AMD EPYC 7763，拥有128个核心。
  • 内存： 1 TB RAM。
  • GPU： 2个AMD MI210。

开始使用

• AMD GPU软件栈：
  • AMD GPU使用的软件栈被称为ROCm（Radeon Open Compute platforM或Radeon Open ECosystem）。没有专门的环境模块，但你可以考虑将以下内容添加到你的~/.bashrc文件中：

    bashCopy code

    export PATH=/opt/rocm-5.4.1/bin:$PATH export LD_LIBRARY_PATH=/opt/rocm-5.4.1/lib:$LD_LIBRARY_PATH

  • 上述目录包含了一些AMD GPU的工具和可执行文件。

常用工具

• rocm-smi：

  • 类似于nvidia-smi，用于管理AMD ROCm系统的管理接口。
  • 示例：

    bashCopy code

    $ rocm-smi

• HIP编程模型：

  • "Heterogeneous-Compute Interface for Portability"，提供适用于大多数代码的C++语法。
  • hipcc是C++编译器，示例：

    bashCopy code

    $ hipcc --version

示例

• Hello World示例：

  • 下载并编译Hello World示例：

    bashCopy code

    $ mkdir test && cd test

  • $ wget https://raw.githubusercontent.com/ROCm-Developer-Tools/HIP-Examples/master/HIP-Examples-Applications/HelloWorld/Makefile

  • $ wget https://raw.githubusercontent.com/ROCm-Developer-Tools/HIP-Examples/master/HIP-Examples-Applications/HelloWorld/HelloWorld.cpp

  • $ make

  • $ ./HelloWorld

• 使用HIP构建LAMMPS：

  • 示例脚本用于构建LAMMPS，确保先下载LAMMPS源代码：

    bashCopy code

    $ bash build_lammps.sh

• TensorFlow示例：

  • 使用Singularity容器运行TensorFlow MNIST示例：

    bashCopy code

    $ git clone https://github.com/PrincetonUniversity/slurm_mnist

  • $ cd slurm_mnist

  • $ singularity exec --rocm

  • $HOME/software/tensorflow_rocm5.4.1-tf2.10-dev.sif python3 download_mnist.py

  • $ singularity exec --rocm

  • $HOME/software/tensorflow_rocm5.4.1-tf2.10-dev.sif python3 mnist_classify.py

• GROMACS示例：

  • 使用Singularity容器运行GROMACS：

    bashCopy code

    $ singularity pull docker://amdih/gromacs:2022.3.amd1_174

  • $ wget ftp://ftp.gromacs.org/pub/benchmarks/rnase_bench_systems.tar.gz

  • $ tar zxvf rnase_bench_systems.tar.gz

  • $ cd rnase_cubic $ singularity exec --rocm ../gromacs_2022.3.amd1_174.sif gmx grompp -f pme_verlet.mdp -c conf.gro -p topol.top -o bench.tpr

  • $ singularity exec --rocm ../gromacs_2022.3.amd1_174.sif gmx mdrun -nsteps 100000 -ntmpi 1 -ntomp 16 -update gpu -s bench.tpr -pin on

有用链接

• AMD ROCm: HIP编程指南
• rocBLAS
• AMD Infinity Hub
• Docker Hub上的ROCm容器
• Singularity页面

以上示例和链接旨在帮助你在della-milan节点上使用AMD MI210 GPU进行编程和性能测试。在实践中，请根据你的具体需求和环境进行相应调整。