Update Dockerfile
Browse files- Dockerfile +6 -0
Dockerfile
CHANGED
|
@@ -49,6 +49,12 @@ RUN apt-get update && apt-get install -y \
|
|
| 49 |
apt-get update && apt-get install -y libcudnn8=8.9.3.* libcudnn8-dev=8.9.3.* && \
|
| 50 |
rm -rf /var/lib/apt/lists/*
|
| 51 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
# Verify cuDNN version
|
| 53 |
RUN dpkg-query -W libcudnn8
|
| 54 |
|
|
|
|
| 49 |
apt-get update && apt-get install -y libcudnn8=8.9.3.* libcudnn8-dev=8.9.3.* && \
|
| 50 |
rm -rf /var/lib/apt/lists/*
|
| 51 |
|
| 52 |
+
# Install specific version of CuDNN
|
| 53 |
+
RUN apt-get update && apt-get install -y \
|
| 54 |
+
libcudnn8=8.9.3.* \
|
| 55 |
+
libcudnn8-dev=8.9.3.* \
|
| 56 |
+
&& rm -rf /var/lib/apt/lists/*
|
| 57 |
+
|
| 58 |
# Verify cuDNN version
|
| 59 |
RUN dpkg-query -W libcudnn8
|
| 60 |
|