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 24, 2023
1 parent 117b384 commit d3cd58e
Showing 1 changed file with 10 additions and 10 deletions.
20 changes: 10 additions & 10 deletions sims/Timeloop/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -26,17 +26,17 @@ 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
RUN echo which conda
RUN conda env create -f environment.yml
RUN apt-get update && apt-get -y install libgmp-dev gcc g++
RUN echo "which conda"
# 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 d3cd58e

Please sign in to comment.