lammpscn icon indicating copy to clipboard operation
lammpscn copied to clipboard

Ubuntu下安装GPU加速的lammps

Open hcb8174875 opened this issue 4 years ago • 4 comments

Ubuntu下安装GPU加速的lammps

Author:黄诚斌

  1. 安装NVIDIA驱动 查询电脑显卡型号,根据维基百科(https://en.wikipedia.org/wiki/CUDA#GPUs_supported) 上的信息查找自己显卡对应的算力,以及该算力对应的CUDA版本。 然后查询该CUDA版本对应的NVIDIA驱动版本,安装对应的驱动版本后执行: nvidia-smi, 如果成功输出则表示驱动安装成功。

  2. 安装CUDA 确定需要安装的CUDA版本后,(https://developer.nvidia.com/cuda-toolkit-archive) 在这个网站上选择对应的版本进行安装,具体步骤为:Linux---x86_64---Ubuntu---18.04---deb(local),然后根据网站给出的安装的步骤一步步进行安装。 安装完毕后在根目录下执行: gedit .bashrc,在末尾添加一行新内容: export PATH=$PATH:/usr/local/cuda-10.2/bin (因为我安装的为10.2,此处根据自己安装的版本进行调整),保存文件后执行: nvcc -V,如果输出正常,即代表安装成功。

  3. 修改文件并进行编译 $ cd lammps/lib/gpu $ vim Makefile.linux

    文件中主要更改内容为: CUDA_HOME:我这里CUDA_HOME 的地址其实是 /usr/local/cuda-10.2 CUDA_ARCH:这里不同的数字对应不同的GPU架构,根据步骤1查找的算力,消除对应的架构之前的 # 即可。 CUDA_PRECISION:这里代表GPU计算过程中的精度,我填的是 -D_SINGLE_SINGLE。 更改完之后执行: $make -f Makefile.linux 如果之后更改Makefile.linux 中的内容之后需要重新编译需要先执行: $make -f Makefile.linux clean

  4. 编译lammps 在lammps/src文件下安装gpu这个package,其余package根据自己需要进行安装。

   $ cd lammps/src
   $ make yes-gpu
   Installing package gpu
   $ make mpi

如果之前编译过lammps,先执行: make clean-all,再进行编译

  1. 测试 可以随便找一个example进行测试,这里我选取的是shear这个例子进行测试 $ mpirun -np 8 lmp_mpi -in in.shear 无GPU $ mpirun -np 8 lmp_mpi -sf gpu -pk gpu 1 -in in.shear 有GPU

本教程参考了卫振宇的帖子 https://madwayne.com/lammps-gpu-acc.html

hcb8174875 avatar Nov 17 '20 13:11 hcb8174875

可以考虑提交 PR 到 README 中。

@Roy-Kid

smartepsh avatar Dec 22 '20 08:12 smartepsh

可以考虑提交 PR 到 README 中。

@Roy-Kid

第一次在github上面提交内容,还不太会用

hcb8174875 avatar Dec 27 '20 06:12 hcb8174875

@hcb8174875 可以在和 @Roy-Kid 沟通下文档的结构,然后提交到正式的代码中。如果是 markdown 或者 org-mode 文件,github 会自动渲染的,这样子其他用户不需要进入 issues 列表来翻找了。

smartepsh avatar Dec 27 '20 15:12 smartepsh

太感谢了,初次接触Ubuntu配置环境实属不易

Antagski avatar Mar 12 '24 14:03 Antagski