Merge pull request Azure#262 from linkernetworks/build/gpu-cuda-lib

build: libcudnn library for GPU container
linkernetworks · May 21, 2023 · d594f1f · d594f1f
2 parents 30dd865 + 0a8a018
commit d594f1f
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 1 deletion.
diff --git a/src/edge/EdgeSolution/modules/kanai/dockerfiles/Dockerfile.gpuamd64 b/src/edge/EdgeSolution/modules/kanai/dockerfiles/Dockerfile.gpuamd64
@@ -4,10 +4,12 @@
 #FROM python:3.8-slim-buster
 FROM nvidia/cuda:11.7.0-runtime-ubuntu20.04
 
-COPY kanai/libcudnn/* /usr/local/cuda/lib64/
+# COPY kanai/libcudnn/* /usr/local/cuda/lib64/
 
 RUN apt update
 RUN apt install ffmpeg libsm6 libxext6 -y
+RUN apt install libcudnn8=8.5.0.96-1+cuda11.7
+RUN cp /usr/lib/x86_64-linux-gnu/*cudnn* /usr/local/cuda/lib64/
 RUN apt install wget unzip -y
 RUN apt install python3.8 pip -y
 RUN pip install -U pip

diff --git a/src/edge/EdgeSolution/modules/kanai/requirements-gpu.txt b/src/edge/EdgeSolution/modules/kanai/requirements-gpu.txt
@@ -14,3 +14,4 @@ httpx
 azure-iot-device
 pytz
 grpcio
+protobuf==3.20.*
diff --git a/src/edge/EdgeSolution/modules/kanai/requirements-jetson.txt b/src/edge/EdgeSolution/modules/kanai/requirements-jetson.txt
@@ -11,3 +11,4 @@ httpx
 azure-iot-device
 pytz
 grpcio
+protobuf==3.19.*
diff --git a/src/edge/EdgeSolution/modules/kanai/requirements-openvino.txt b/src/edge/EdgeSolution/modules/kanai/requirements-openvino.txt
@@ -11,3 +11,4 @@ httpx
 azure-iot-device
 pytz
 grpcio
+protobuf==3.19.*
diff --git a/src/edge/EdgeSolution/modules/kanai/requirements.txt b/src/edge/EdgeSolution/modules/kanai/requirements.txt
@@ -14,3 +14,4 @@ httpx
 azure-iot-device
 pytz
 grpcio
+protobuf==3.20.*
-Original file line number
+Diff line change
@@ Expand Up / @@ -14,3 +14,4 @@ httpx @@
     azure-iot-device
     pytz
     grpcio
+    protobuf==3.20.*