update dockerfile

docker/Dockerfile

@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.9.1-cuda11.1-cudnn8-runtime
+FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime
 
 RUN apt-get update -y && DEBIAN_FRONTEND=noninteractive apt-get install software-properties-common -y && \
     add-apt-repository -y multiverse && apt-get update -y && apt-get upgrade -y && \
@@ -8,14 +8,6 @@ RUN apt-get update -y && DEBIAN_FRONTEND=noninteractive apt-get install software
 RUN apt-get update -y && apt-get install -y curl libssl-dev openssl libopenblas-dev \
     libhdf5-dev hdf5-helpers hdf5-tools libhdf5-serial-dev libprotobuf-dev protobuf-compiler git
 
-RUN pip install ray==1.8.0 && pip install ray[tune] && pip install ray[rllib]
+RUN pip install --upgrade pip && pip install marllib
 
-RUN git clone https://github.com/Replicable-MARL/MARLlib.git
-
-WORKDIR /workspace/MARLlib
-
-RUN pip install -e .
-
-RUN pip install icecream && pip install lbforaging==1.0.15 && pip install supersuit && pip install gym==0.21.0 && pip install importlib-metadata==4.13.0
-
-RUN python patch/add_patch.py -y
+RUN pip install protobuf==3.17 && pip install gym==0.20.0

docker/build.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
 
-echo 'Building Dockerfile with image name marl:1.0'
-#docker build -t marl:1.0 .
-docker build --no-cache -t marl:1.0 . # build docker from nothing
+echo 'Building Dockerfile with image name marllib:1.0'
+#docker build -t marllib:1.0 .
+docker build --no-cache -t marllib:1.0 . # build docker from nothing