Skip to content

Commit

Permalink
Removed processing with GPU
Browse files Browse the repository at this point in the history
  • Loading branch information
murilosb committed Aug 8, 2023
1 parent 3d2d963 commit 2ae2b3e
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions Dockerfile.dependencies
Original file line number Diff line number Diff line change
@@ -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

Expand Down Expand Up @@ -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 \
Expand Down

0 comments on commit 2ae2b3e

Please sign in to comment.