Fix Dockerfile

ispras · Dec 25, 2023 · 8d2b451 · 8d2b451
1 parent c4fe41e
commit 8d2b451
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -6,8 +6,6 @@ ENV RESOURCES_PATH "/dedoc_root/resources"
 
 ADD requirements.txt .
 RUN pip3 install --no-cache-dir -r requirements.txt
-RUN pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html
-
 
 RUN mkdir /dedoc_root
 RUN mkdir /dedoc_root/dedoc

diff --git a/docker_gpu/README.md b/docker_gpu/README.md
@@ -9,7 +9,7 @@ cd docker_gpu
 docker-compose up --build
 ```
 
-You can change index if CUDA device at `docker-compose.yml`:
+You can change index of CUDA device at `docker-compose.yml`:
 ```
 NVIDIA_VISIBLE_DEVICES: 0
 NVIDIA_VISIBLE_DEVICES: 0, 3