Skip to content

Commit

Permalink
minor conda
Browse files Browse the repository at this point in the history
  • Loading branch information
a-saraf committed Oct 25, 2023
1 parent 9559729 commit 9706eca
Showing 1 changed file with 9 additions and 10 deletions.
19 changes: 9 additions & 10 deletions sims/Timeloop/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -26,17 +26,16 @@ RUN chown -R 1000:root /home/workspace && chmod -R 775 /home/workspace
RUN git clone --recursive https://github.com/srivatsankrishnan/oss-arch-gym.git
RUN chown -R 1000:root /home/workspace && chmod -R 775 /home/workspace
RUN cd oss-arch-gym
SHELL ["which", "conda"]
# RUN conda env create -f environment.yml
# RUN apt-get update && apt-get -y install libgmp-dev gcc g++
RUN conda env create -f environment.yml
RUN apt-get update && apt-get -y install libgmp-dev gcc g++

# # Make RUN commands use the new environment:
# RUN echo "conda activate oss-arch-gym" >> ~/.bashrc
# SHELL ["conda", "run", "-n", "oss-arch-gym", "/bin/bash", "--login", "-c"]
# Make RUN commands use the new environment:
RUN echo "conda activate oss-arch-gym" >> ~/.bashrc
SHELL ["conda", "run", "-n", "oss-arch-gym", "/bin/bash", "--login", "-c"]

# RUN cd oss-arch-gym/acme && pip install .[jax,tf,testing,envs] && pip install envlogger[tfds] && cd ../ && pip install -e . && pip install absl-py && pip install scikit-optimize
# ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/bin/conda/envs/oss-arch-gym/lib
# RUN echo $LD_LIBRARY_PATH
RUN cd oss-arch-gym/acme && pip install .[jax,tf,testing,envs] && pip install envlogger[tfds] && cd ../ && pip install -e . && pip install absl-py && pip install scikit-optimize
ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/bin/conda/envs/oss-arch-gym/lib
RUN echo $LD_LIBRARY_PATH

# ENTRYPOINT ["conda", "run", "-n", "oss-arch-gym", "python", "/home/workspace/src/timeloop-examples/workspace/final-project/example_designs/eyeriss_like/arch-gym/sims/Timeloop/launch_gcp.py"]
ENTRYPOINT ["conda", "run", "-n", "oss-arch-gym", "python", "/home/workspace/src/timeloop-examples/workspace/final-project/example_designs/eyeriss_like/arch-gym/sims/Timeloop/launch_gcp.py"]

0 comments on commit 9706eca

Please sign in to comment.