Skip to content

Instantly share code, notes, and snippets.

@link89
Last active May 10, 2024 01:49
Show Gist options
  • Save link89/add281c6a389d67ee0631685a307c036 to your computer and use it in GitHub Desktop.
Save link89/add281c6a389d67ee0631685a307c036 to your computer and use it in GitHub Desktop.
DeepMD-kit: conda installation

2.2.8

CONDA_OVERRIDE_CUDA="11.8" conda create -p deepmd/2.2.8/gpu deepmd-kit=2.2.8=cuda118py310*  lammps horovod -c conda-forge 

To work around a known issue of conda-forge's tensorflow package, one has to run pip install nvidia-cuda-nvcc-cu11 in the conda environment.

Before 2.2.8

CONDA_OVERRIDE_CUDA="11.6" conda create -n deepmd deepmd-kit=2.2.7=*gpu libdeepmd=2.2.7=*gpu lammps cudatoolkit=11.6 horovod -c https://conda.deepmodeling.com -c defaults
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment