add cuda

Dockerfile

@@ -15,7 +15,7 @@ COPY pyproject.toml poetry.lock /app/
 # Install poetry
 RUN pip install --upgrade pip
 RUN pip install poetry
-RUN pip install torch==1.10.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
+RUN pip install torch==1.11.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
 
 
 # Set environment variable to create virtualenv within the project directory