Nikita commited on
Commit
936da19
·
1 Parent(s): dd32739

dockerfile to use .yaml file, requirements.txt didnt work

Browse files
Files changed (2) hide show
  1. Dockerfile +26 -0
  2. environment.yaml +34 -0
Dockerfile ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Base image with Conda
2
+ FROM continuumio/miniconda3
3
+
4
+ # Set working directory
5
+ WORKDIR /app
6
+
7
+ # Copy environment.yaml for conda
8
+ COPY environment.yaml /tmp/environment.yaml
9
+
10
+ # Create the Conda environment
11
+ RUN conda env create -f /tmp/environment.yaml
12
+
13
+ # Set default shell to use conda env
14
+ SHELL ["conda", "run", "--no-capture-output", "-n", "tirex", "/bin/bash", "-c"]
15
+
16
+ # Copy all project files and folders
17
+ COPY app.py /app/
18
+ COPY static /app/static
19
+ COPY data /app/data
20
+ COPY tirex /app/tirex
21
+
22
+ # Expose the default port
23
+ EXPOSE 7860
24
+
25
+ # Run your app
26
+ CMD ["python", "app.py"]
environment.yaml ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: tirex
2
+ channels:
3
+ - pytorch
4
+ - nvidia
5
+ - conda-forge
6
+ - defaults
7
+ dependencies:
8
+ - cuda=12.6
9
+ - cuda-nvcc=12.6
10
+ - gxx_linux-64=13.2.0
11
+ - python=3.11
12
+ - pip
13
+ - cmake
14
+ - ninja
15
+ - cuda-toolkit=12.6
16
+ - cuda-cccl=12.6
17
+ - pip:
18
+ - --index-url https://download.pytorch.org/whl/cu126
19
+ - --extra-index-url https://pypi.org/simple
20
+ - torch~=2.6.0
21
+ - torchvision~=0.21.0
22
+ - torchaudio~=2.6.0
23
+ - pre-commit
24
+ - xlstm~=2.0.3
25
+ - einops~=0.8.1
26
+ - huggingface-hub~=0.32.0
27
+ - lightning~=2.5.1
28
+ - numpy~=2.1.3
29
+ - pandas~=2.2.3
30
+ - dacite~=1.9.2
31
+ - pytest
32
+ - gradio
33
+ - plotly
34
+ - pillow