How to run dpgen model_devi on RTX A6000 (CUDA 11+) when lammps-dp conda packages are CUDA 10 only?

[Collapsar126]

Hi,

I’m trying to run the model_devi stage of dpgen with GPU DeepMD on a
workstation with RTX A6000 (compute 8.6, CUDA 11+) and ran into a
compatibility issue.

• Training uses a conda env with deepmd-kit CUDA 11.6 GPU build and works fine.
• For model_devi, I created another env with
  deepmd-kit + lammps-dp CUDA 10.1 GPU builds from conda.deepmodeling.com.

When I run LAMMPS with pair_style deepmd, TensorFlow/DeepMD prints an error like:

CUDA runtime implicit initialization on GPU:0 failed.
Status: device kernel image is invalid

nvidia-smi shows an RTX A6000, so I suspect the CUDA 10.1 build of lammps-dp
is not compatible with this GPU.

mamba search shows CUDA 11 builds for deepmd-kit, but I can’t find any
CUDA 11 lammps-dp packages on the deepmodeling channel.

Questions:

1. What is the recommended way to run dpgen’s model_devi on RTX A6000?

   • Use CPU-only lammps-dp for model_devi? That will take much more time
   • Or manually compile LAMMPS + DeePMD against CUDA 11?

2. Are there plans to provide CUDA 11+ lammps-dp conda packages for newer GPUs?

Thanks a lot for any suggestions!