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.
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