[SE3Transformer/PyT] Fix requirements on newer containers
This commit is contained in:
parent
58df33d1a2
commit
72a90514f2
|
@ -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
|
||||
|
|
Loading…
Reference in a new issue