将deepmd的数据集转换成GPUMD的数据集并训练(包含GPUMD的安装)
首先安装GPUMD需要自己装一个CUDA,这里我安装了CUDA11.5,同时安装过程中需要注意只安装toolkit就可以了,其他的不需要安装,注意在options修改安装路径,之后再GPUMD/src目录下修改makefile(主要修改sm-86就行),修改完直接make就能安装好。
首先在github找到deep2nep.py的脚本,将dp的数据集按照例子要求存放,然后python nep2xyz.py deepmd
完成将会得到train.in,但是训练需要将它分成训练集和测试机,这里我们需要用到樊老师的split_train.cpp脚本,输入命令g++ -std=c++11 -O3 split_train.cpp,会得到a.out文件,然后输入命令./a.out test_size(test_size需要自己设置),最后还需要使用nep2xyz.py脚本,分别将训练集和测试集转换一下,就可以训练nep了!
评论
发表评论