diff --git a/setup/install-gpu.sh b/setup/install-gpu.sh index f87845c6c..e40ec4172 100755 --- a/setup/install-gpu.sh +++ b/setup/install-gpu.sh @@ -29,7 +29,7 @@ conda upgrade -y --all # install and configure theano pip install theano echo "[global] -device = gpu +device = cuda floatX = float32 [cuda]