Set ray==1.0.1 in Dockerfile

HazyResearch · Nov 10, 2020 · a5d7ca6 · a5d7ca6
1 parent a89fe2f
commit a5d7ca6
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/convolution/docker/Dockerfile b/convolution/docker/Dockerfile
@@ -38,7 +38,5 @@ RUN conda install -y -c pytorch cudatoolkit=10.1 pytorch=1.7.0 torchvision \
 
 # Other libraries
 # wanbd>=0.10.0 tries to read from ~/.config, and that causes permission error on dawn
-RUN pip install pytorch-lightning==1.0.3 pytorch-lightning-bolts==0.2.5 ray[tune]==1.0.0 hydra-core==1.0.3 wandb==0.9.7 munch scikit-learn \
-    && rm -rf /home/user/.cache/pip
-RUN pip install -U https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-1.1.0.dev0-cp38-cp38-manylinux1_x86_64.whl \
+RUN pip install pytorch-lightning==1.0.3 pytorch-lightning-bolts==0.2.5 ray[tune]==1.0.1 hydra-core==1.0.3 wandb==0.9.7 munch scikit-learn \
     && rm -rf /home/user/.cache/pip
diff --git a/convolution/train.py b/convolution/train.py
@@ -14,7 +14,6 @@
 import datamodules
 import tasks
 from pl_runner import pl_train
-from utils import to_scalar, dictconfig_to_munch
 from tee import StdoutTee, StderrTee