Update Dockerfile
Browse files- Dockerfile +7 -0
Dockerfile
CHANGED
|
@@ -13,6 +13,13 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/
|
|
| 13 |
|
| 14 |
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
|
| 15 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
|
| 17 |
USER root
|
| 18 |
|
|
|
|
| 13 |
|
| 14 |
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
|
| 15 |
|
| 16 |
+
RUN echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" > /etc/apt/sources.list.d/cuda.list
|
| 17 |
+
|
| 18 |
+
# Install FFmpeg
|
| 19 |
+
|
| 20 |
+
RUN apt-get update && \
|
| 21 |
+
apt-get install -y python3-pip ffmpeg libcublas-11-8 libcudnn8=8.6.0.163-1+cuda11.8
|
| 22 |
+
#libcudnn8=8.8.0.121-1+cuda11.8
|
| 23 |
|
| 24 |
USER root
|
| 25 |
|