From 2ae2b3ea4b0a8f063ff0940dd7bc893318a1e42d Mon Sep 17 00:00:00 2001 From: Murilo Santos Bueno Date: Tue, 8 Aug 2023 16:28:23 -0300 Subject: [PATCH] Removed processing with GPU --- Dockerfile.dependencies | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/Dockerfile.dependencies b/Dockerfile.dependencies index 9af6146..362cd17 100644 --- a/Dockerfile.dependencies +++ b/Dockerfile.dependencies @@ -1,4 +1,5 @@ -FROM nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu22.04 +# FROM nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu22.04 +FROM ubuntu:22.04 ARG PKGTMPDIR=/tmp/pkgs @@ -242,7 +243,7 @@ RUN mkdir -p ${PKGTMPDIR} && cd ${PKGTMPDIR} \ && python -m pip install cmake \ && python -m pip install numpy \ && python -m pip install jep==4.0.3 \ - && python -m pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117 \ + # && python -m pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117 \ && echo "Due to container high storage usage, Tensorflow was removed - it's only used on yahoo image detection - if you want to use it, uncomment the following line:" \ && echo "python -m pip install tensorflow==2.9.1 tensorrt==8.5.3.1" \ && python -m pip install keras \