!pip install deepmd-kit[gpu,cu11,ipi]

from deepmd.infer import DeepPot

%%bash
cat <<EOT > run.py
from deepmd.infer import DeepPot
EOT

!python run.py