From 999a953da2c72d1cfa46c791eeddb093282e3413 Mon Sep 17 00:00:00 2001 From: Timotej979 Date: Sun, 14 Jan 2024 11:40:12 +0100 Subject: [PATCH 1/2] Remove double os import, add missing cv2 import --- src/inference/inference_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/inference/inference_image.py b/src/inference/inference_image.py index 4684ae3..0321f5f 100644 --- a/src/inference/inference_image.py +++ b/src/inference/inference_image.py @@ -1,4 +1,5 @@ import os +import cv2 import torch import torch.nn as nn import torch.nn.functional as F @@ -7,7 +8,6 @@ from torchvision import datasets,transforms,models,utils import numpy as np import matplotlib.pyplot as plt -import os import pandas as pd from PIL import Image import sys From 4c9cea9898c53bfc52b3e55d38092c4ea63a41b2 Mon Sep 17 00:00:00 2001 From: Timotej979 Date: Sun, 14 Jan 2024 11:41:09 +0100 Subject: [PATCH 2/2] Update CUDA image to an existing version --- dockerfiles/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dockerfiles/Dockerfile b/dockerfiles/Dockerfile index 0709647..b7e5ff8 100644 --- a/dockerfiles/Dockerfile +++ b/dockerfiles/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:11.4.1-runtime-ubuntu20.04 +FROM nvidia/cuda:11.4.3-runtime-ubuntu20.04 RUN apt-get -y update && \