DeepLearningExamples/PyTorch/DrugDiscovery/SE3Transformer/requirements.txt

5 lines
93 B
Plaintext

e3nn==0.3.3
wandb==0.12.0
pynvml==11.0.0
git+https://github.com/NVIDIA/dllogger#egg=dllogger