Skip to content

Commit

Permalink
Merge pull request #127 from hmy3743/master
Browse files Browse the repository at this point in the history
fix docker setting
  • Loading branch information
pengsida committed May 8, 2020
2 parents 5b395e1 + be135c0 commit 7db2985
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 2 deletions.
5 changes: 4 additions & 1 deletion docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64
ENV PYTHON_VERSION=3.6.7

RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
RUN curl -L -o ~/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
chmod +x ~/miniconda.sh && \
~/miniconda.sh -b -p /opt/conda && \
rm ~/miniconda.sh && \
Expand Down Expand Up @@ -65,4 +65,7 @@ RUN source activate pvnet \
&& python build_extend_utils_cffi_for_docker.py
ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/$username/pvnet/lib/utils/extend_utils/lib

# set up project root environment variable
ENV root=/home/pvnet/pvnet

WORKDIR /home/$username
2 changes: 1 addition & 1 deletion docker/launch-docker.sh
Original file line number Diff line number Diff line change
Expand Up @@ -6,5 +6,5 @@ docker run -it --rm \
--name pvnet-nvidia-c \
--net host \
--privileged \
--runtime=nvidia \
--gpus all \
pvnet-nvidia

0 comments on commit 7db2985

Please sign in to comment.