[SE3Transformer/PyT] Fix requirements on newer containers

This commit is contained in:
Alexandre Milesi 2021-08-29 02:24:20 -07:00 committed by Krzysztof Kudrynski
parent 58df33d1a2
commit 72a90514f2

View file

@ -1,2 +1,4 @@
e3nn==0.3.3
wandb==0.12.0
wandb==0.12.0
pynvml==11.0.0
git+https://github.com/NVIDIA/dllogger#egg=dllogger