安装miniconda3 bash Miniconda3-latest-Linux-x86_64.sh 一直enter即可 输入yes
创建conda虚拟环境:conda create -n jax-env python=3.8
下载源码或者复制过去 git clone https://kgithub.com/cagrikymk/Jax-ReaxFF
进入源码目录 cd /jax/Jax-ReaxFF (根据相应目录更改)
激活 conda activate jax-env
安装python对于的包 pip install .
如果不行,需要对应安装包,依次安装下面的包,不能超过它的范围

pip3 install --upgrade jax==0.3.0 jaxlib==0.1.74 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip3 install --upgrade numba==0.51.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip3 install --upgrade tabulate==0.8.9 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip3 install --upgrade numpy==1.18.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip3 install --upgrade numpy-1.21.6-cp38-cp38-manylinux_2_12_x86_64.manylinux2010_x86_64.whl
-i https://pypi.tuna.tsinghua.edu.cn/simple
pip3 install --upgrade numpy==1.21.6 -i https://pypi.tuna.tsinghua.edu.cn/simple
可以使用conda list 查看是否安装成功
创建要计算的目录,cd 进入目录,使用以下CPU命令版本:
jaxreaxff --init_FF Datasets/cobalt/ffield_lit \
--params Datasets/cobalt/params \
--geo Datasets/cobalt/geo \
--train_file Datasets/cobalt/trainset.in \
--num_e_minim_steps 200 \
&nbs


1257

被折叠的 条评论
为什么被折叠?



