怎么在anaconda中下载jax是cup的好还是GPU的好

Question

我想要下载dmff软件包，介绍里让我下载这个jax，我先在是在anaconda下Python中应用，我试了CPU的按它的命令下载不出来，麻烦知道怎么办的教一下，谢谢！

Answer 1

引自免费微信小程序：皆我百晓生

看起来你正在尝试安装DMFF软件包，并且遇到了一些问题。首先，让我们解决你在安装JAX时遇到的问题。根据DMFF的安装指南，你需要选择CPU或GPU版本的JAX。你之前尝试的是CPU版本，但似乎有些错误。

为了确保正确安装，按照以下步骤操作：

1. 首先，你需要激活你的名为dmff的Conda环境。如果你还没有创建这个环境，可以使用以下命令创建它：

conda create -n dmff python=3.9 --yes

然后激活它：

conda activate dmff

1. 确保你已经正确安装了Conda和Python。在激活的环境中，你可以通过输入python --version和conda --version检查它们的版本。

2. 安装JAX的CPU版本。在命令行中输入：

pip install "jax[cpu]==0.4.14"

1. 接下来，按照安装指南安装其他依赖项。对于mdtraj、optax、jaxopt和pymbar，使用以下命令：

conda install -c conda-forge mdtraj==1.9.7
pip install optax==0.1.3 pymbar==4.0.1

1. 对于OpenMM，使用Conda安装：

conda install -c conda-forge openmm==7.7.0

1. 最后，你可以使用Git从GitHub克隆DMFF项目。打开一个新的终端窗口（或者在当前终端窗口中确保你已经在dmff环境中），然后执行以下命令：

git clone https://github.com/deepmodeling/DMFF.git
cd DMFF

1. 在DMFF目录中，安装所需的库：

pip install -r requirements.txt

完成上述步骤后，你应该成功安装了DMFF并准备开始使用它。如果你在任何步骤中遇到问题，请告诉我具体错误信息，以便我能提供更详细的指导。