Commit 3f358710 authored by Siddhartha Laghuvarapu's avatar Siddhartha Laghuvarapu
Browse files

Update Dockerfile to add torch dependencies

parent 693b06c0
......@@ -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}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment