Unverified Commit deec6dea authored by Hang Zhang's avatar Hang Zhang Committed by GitHub
Browse files

ADD Docker (#305)

* add docker
parent e219aa7e
FROM nvcr.io/nvidia/pytorch:19.05-py3
# Set working directory # Set working directory
WORKDIR /workspace
#ENV PYTHONPATH "${PYTHONPATH}:/workspace"
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y python3-tk python-pip git tmux htop tree
RUN python -m pip install --upgrade pip
RUN python -m pip install torch==1.4.0
RUN python -m pip install torchvision==0.5.0
RUN python -m pip install pycocotools==2.0.0
#RUN chmod a+rwx -R /opt/conda/
COPY ./setup.py .
COPY ./encoding ./encoding
RUN python setup.py develop
COPY ./experiments ./experiments
docker build -t encoding .
docker run --rm -it --ulimit memlock=-1 --ulimit stack=67108864 -v $COCO_DIR:/coco --ipc=host --gpus=all encoding
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