Update Dockerfile to add torch dependencies

Dockerfile

@@ -52,6 +52,8 @@ RUN ln -s neural-mmo/forge forge \
 RUN conda install cmake -y && conda clean -y -a
 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}