Skip to content

Commit

Permalink
docker replace conda to pip, fix path setup
Browse files Browse the repository at this point in the history
  • Loading branch information
emotionor committed Jun 25, 2024
1 parent 019dfc5 commit be4de56
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
1 change: 1 addition & 0 deletions .github/workflows/docker.yml
Original file line number Diff line number Diff line change
Expand Up @@ -70,6 +70,7 @@ jobs:
env:
FLASH_ATTENTION_SKIP_CUDA_BUILD: "TRUE"
run: |
cd unimol_tools
python setup.py sdist --dist-dir=dist
- name: Deploy
Expand Down
4 changes: 2 additions & 2 deletions unimol/docker/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
FROM dptechnology/unicore:0.0.1-pytorch1.11.0-cuda11.3

RUN conda install -y -c conda-forge rdkit==2021.09.5 && conda clean -ya
RUN pip install rdkit-pypi==2021.9.5.1

RUN ldconfig && \
apt-get clean && \
apt-get autoremove && \
rm -rf /var/lib/apt/lists/* /tmp/* && \
conda clean -ya
pip cache purge

0 comments on commit be4de56

Please sign in to comment.