Dell AMD EPYC 7662 / NVIDIA Tesla A100

Softwares Instalados

Categoria: Programação

DeepMD-Kit
Versão 2.1.3
Descrição DeepMD-Kit is a package written in Python/C++, designed to minimize the effort required to build deep learning based model of interatomic potential energy and force field and to perform molecular dynamics (MD). This brings new hopes to addressing the accuracy-versus-efficiency dilemma in molecular simulations. Applications of DeePMD-kit span from finite molecules to extended systems and from metallic systems to chemically bonded systems.
Documentação Documentação
Licença licensed under GNU LGPLv3.0
Localização https://deepmodeling.com/

Como usar

- Configuração da conta para uso:

source /opt/pub/deepmd-kit/2.1.3/deepmd-kit-2.1.3.sh

- Executáveis disponíveis na pasta:

/opt/pub/spack/anaconda3/2021.05/intel/2022.0.1/envs/deepmd/bin

Exemplo de script para submissao e processamento em filas do PBS (Fila: duasgpus)

#!/bin/bash
### Nome da fila de execucao ###
#PBS -q duasgpus

### Nome do job ###
#PBS -N teste

### Nome do arquivo de saida ###
#PBS -o saida

### Nome do arquivo de erros ###
#PBS -e erro

### Recursos necessarios para execucao:
###   1 node, 64 cpus, 1 gpu (fila umagpu) 
###   ou 1 node, 128 cpus, 2 gpus (fila duasgpus)
### Nao eh necessario especificar diretiva -l nodes=X,ppn=Y.
###  Numero de cpus eh configurado automaticamente de acordo com a fila

cd $PBS_O_WORKDIR
echo "-----------------------------------------"
echo "Inicio do job:" `date`

source /opt/pub/deepmd-kit/2.1.3/deepmd-kit-2.1.3.sh

export OMP_NUM_THREADS=2
export TF_INTRA_OP_PARALLELISM_THREADS=2
export TF_INTER_OP_PARALLELISM_THREADS=1

dp train input.json

echo "Final do job:" `date`
echo "-----------------------------------------"