Update Dockerfile

Dockerfile

 # Select one of the images based on your case
 # FROM aicrowd/base-images:neural-mmo-py38-torch-190
 # FROM aicrowd/base-images:neural-mmo-py38-torch-180
-# FROM aicrowd/base-images:neural-mmo-py38-torch-171
-FROM aicrowd/base-images:neural-mmo-py38
+FROM aicrowd/base-images:neural-mmo-py38-torch-171
+# FROM aicrowd/base-images:neural-mmo-py38
 
 COPY --chown=1001:1001 requirements.txt ${HOME_DIR}/requirements.txt
 RUN pip install -r requirements.txt --no-cache-dir
-RUN conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
 
 
 COPY --chown=1001:1001 . ${HOME_DIR}