From 6aad04287b8d26ca36d843e4645d83f950464d51 Mon Sep 17 00:00:00 2001
From: Faraz Fallahi <fffaraz@gmail.com>
Date: Fri, 9 Apr 2021 20:46:42 -0400
Subject: [PATCH] Update Dockerfile

---
 Dockerfile | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index e8af1ff..fbf9bf8 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -47,6 +47,7 @@ RUN \
 RUN pip3 install --upgrade dlib
 RUN pip3 install --upgrade frida frida-tools
 
+# https://github.com/facebook/prophet/blob/master/python/requirements.txt
 RUN \
     pip3 install --upgrade Cython cmdstanpy==0.9.68 pystan~=2.19.1.1 numpy pandas matplotlib LunarCalendar convertdate holidays setuptools-git python-dateutil tqdm && \
     pip3 install --upgrade fbprophet && \
@@ -85,6 +86,16 @@ RUN \
     ssh-keygen -A && \
     exit 0
 
+RUN \
+    set -eux && \
+    wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin && \
+    mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 && \
+    apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub && \
+    add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /" && \
+    apt-get update && \
+    apt-get -y install cuda && \
+    exit 0
+
 ENV IGNORE_CC_MISMATCH=1
 ENV PATH=$PATH:/usr/local/cuda/bin
 ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64