lammpscn
lammpscn copied to clipboard
Ubuntu下安装GPU加速的lammps
Ubuntu下安装GPU加速的lammps
Author:黄诚斌
-
安装NVIDIA驱动 查询电脑显卡型号,根据维基百科(https://en.wikipedia.org/wiki/CUDA#GPUs_supported) 上的信息查找自己显卡对应的算力,以及该算力对应的CUDA版本。 然后查询该CUDA版本对应的NVIDIA驱动版本,安装对应的驱动版本后执行:
nvidia-smi
, 如果成功输出则表示驱动安装成功。 -
安装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
,如果输出正常,即代表安装成功。 -
修改文件并进行编译
$ 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
-
编译lammps 在lammps/src文件下安装gpu这个package,其余package根据自己需要进行安装。
$ cd lammps/src
$ make yes-gpu
Installing package gpu
$ make mpi
如果之前编译过lammps,先执行: make clean-all
,再进行编译
- 测试
可以随便找一个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
可以考虑提交 PR 到 README 中。
@Roy-Kid
可以考虑提交 PR 到 README 中。
@Roy-Kid
第一次在github上面提交内容,还不太会用
@hcb8174875 可以在和 @Roy-Kid 沟通下文档的结构,然后提交到正式的代码中。如果是 markdown 或者 org-mode 文件,github 会自动渲染的,这样子其他用户不需要进入 issues
列表来翻找了。
太感谢了,初次接触Ubuntu配置环境实属不易