diff --git a/env.yml b/env.yml index 4b6050b84..9eaaffbe7 100644 --- a/env.yml +++ b/env.yml @@ -32,7 +32,7 @@ dependencies: - pytorch >=1.10.2,<2.0 - tensorboard - lightning >=2.0 - - lightning-graphcore + - pip::lightning-graphcore # optional, for using IPUs only - torchvision - torchmetrics >=0.7.0,<0.11 - ogb