YuITC commited on
Commit
b4ad7cb
·
1 Parent(s): 0063d17

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -3
Dockerfile CHANGED
@@ -5,9 +5,11 @@ WORKDIR /app
5
  # Get dependencies
6
  COPY requirements.txt .
7
 
8
- RUN conda install -y python=3.10 \
9
- pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia \
10
- faiss-gpu=1.9.0 -c pytorch -c nvidia && \
 
 
11
  conda clean -afy
12
 
13
  RUN pip install --no-cache-dir -r requirements.txt
 
5
  # Get dependencies
6
  COPY requirements.txt .
7
 
8
+ RUN conda install -y \
9
+ -c pytorch -c nvidia \
10
+ python=3.10 \
11
+ pytorch torchvision torchaudio pytorch-cuda=12.1 \
12
+ faiss-gpu=1.9.0 && \
13
  conda clean -afy
14
 
15
  RUN pip install --no-cache-dir -r requirements.txt