hoonsubin commited on
Commit
bb363fd
·
1 Parent(s): c984e43

update docker

Browse files
Files changed (1) hide show
  1. Dockerfile +14 -5
Dockerfile CHANGED
@@ -11,6 +11,9 @@ RUN apt-get update && apt-get install -y \
11
  libsndfile1-dev \
12
  kmod \
13
  libomp5 \
 
 
 
14
  libopenblas0-pthread \
15
  libgl1 \
16
  libglib2.0-0 \
@@ -22,16 +25,22 @@ RUN apt-get update && apt-get install -y \
22
  build-essential \
23
  && rm -rf /var/lib/apt/lists/*
24
 
 
 
 
 
 
 
 
 
 
 
25
  # Update pip and install base requirements
26
  RUN pip install --upgrade pip setuptools wheel && \
27
  pip install numpy==1.26.4
28
 
29
  # Install CPU-only PyTorch first
30
- RUN pip install --no-cache-dir torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 \
31
- --index-url https://download.pytorch.org/whl/cpu
32
-
33
- # Install CPU-only PyTorch components
34
- RUN pip install torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cpu
35
 
36
  # Install Python requirements with CPU-only constraints
37
  COPY requirements.txt .
 
11
  libsndfile1-dev \
12
  kmod \
13
  libomp5 \
14
+ libssl3 \
15
+ ca-certificates \
16
+ openssl \
17
  libopenblas0-pthread \
18
  libgl1 \
19
  libglib2.0-0 \
 
25
  build-essential \
26
  && rm -rf /var/lib/apt/lists/*
27
 
28
+ # Install Rust compiler
29
+ RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | \
30
+ sh -s -- -y --default-toolchain stable --profile minimal && \
31
+ echo 'export PATH="$HOME/.cargo/bin:$PATH"' >> /etc/profile.d/rust.sh
32
+
33
+ ENV PATH="/root/.cargo/bin:${PATH}"
34
+
35
+ # Verify installation
36
+ RUN rustc --version && cargo --version
37
+
38
  # Update pip and install base requirements
39
  RUN pip install --upgrade pip setuptools wheel && \
40
  pip install numpy==1.26.4
41
 
42
  # Install CPU-only PyTorch first
43
+ RUN pip install torch torchvision torchaudio==2.2.2 --extra-index-url https://download.pytorch.org/whl/cpu
 
 
 
 
44
 
45
  # Install Python requirements with CPU-only constraints
46
  COPY requirements.txt .