Changes-2

Dockerfile

@@ -19,7 +19,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 RUN pip install --upgrade pip
 
 RUN pip install --no-cache-dir -r requirements.txt
-RUN pip install --no-cache-dir -r requirements_gpu.txt
+# RUN pip install --no-cache-dir -r requirements_gpu.txt
 RUN pip install --no-cache-dir -r requirements_app.txt
 
 RUN useradd -m -u 1000 myuser

requirements.txt

@@ -8,8 +8,8 @@ Pillow>=7.1.2
 PyYAML>=5.3.1
 requests>=2.23.0
 scipy>=1.4.1
-# torch>=1.7.0,!=1.12.0
-# torchvision>=0.8.1,!=0.13.0
+torch>=1.7.0,!=1.12.0
+torchvision>=0.8.1,!=0.13.0
 tqdm>=4.41.0
 protobuf<4.21.3
 

requirements_gpu.txt

@@ -2,8 +2,8 @@
 -i https://download.pytorch.org/whl/cu113
 
 # CUDA 11.3
-# torch==1.11.0+cu113
-# torchvision==0.12.0+cu113
+torch==1.11.0+cu113
+torchvision==0.12.0+cu113
 
 # CPU-only version of PyTorch
 # torch==1.11.0
@@ -14,5 +14,5 @@
 # torch==torch-1.11.0-cp310-cp310-linux_x86_64.whl
 # torchvision==torchvision-0.12.0-cp310-cp310-linux_x86_64.whl
 
-torch==torch-1.11.0+cu113-cp310-cp310-linux_x86_64.whl
-torchvision==torchvision-0.12.0+cu113-cp310-cp310-linux_x86_64.whl
+# torch==torch-1.11.0+cu113-cp310-cp310-linux_x86_64.whl
+# torchvision==torchvision-0.12.0+cu113-cp310-cp310-linux_x86_64.whl