Commit 2b5c7548 authored by Siddhartha Laghuvarapu's avatar Siddhartha Laghuvarapu
Browse files

Update Dockerfile

parent eae29a3b
# 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}
......
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