From 864ab4cb30a179f0b86330f186572ca5885ba461 Mon Sep 17 00:00:00 2001
From: Ubuntu <ubuntu@ip-172-31-7-246.us-west-2.compute.internal>
Date: Tue, 17 Apr 2018 06:18:13 +0000
Subject: [PATCH] Fixing install-gpu

---
 setup/install-gpu.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/setup/install-gpu.sh b/setup/install-gpu.sh
index f87845c6c..4d02480df 100755
--- a/setup/install-gpu.sh
+++ b/setup/install-gpu.sh
@@ -27,6 +27,8 @@ conda install -y bcolz
 conda upgrade -y --all
 
 # install and configure theano
+
+conda install mkl=2017
 pip install theano
 echo "[global]
 device = gpu
@@ -35,6 +37,8 @@ floatX = float32
 [cuda]
 root = /usr/local/cuda" > ~/.theanorc
 
+echo "export THEANO_FLAGS=mode=FAST_RUN,device=cuda,floatX=float32" >> ~/.bashrc
+
 # install and configure keras
 pip install keras==1.2.2
 mkdir ~/.keras