diff --git a/1_install_cuda.sh b/1_install_cuda.sh
index a97d2cc9510f6ce73a3757bdd845ff1cc03bc899..051e2bdb43c25bdd94084fae776869133db78a8e 100644
--- a/1_install_cuda.sh
+++ b/1_install_cuda.sh
@@ -1,15 +1,10 @@
 #!/usr/bin/env bash
 # Installation script for CUDA and drivers on Ubuntu, by Giuseppe Attardi (attardi@di.unipi.it)
-# Derived from:
-# https://github.com/deeplearningparis/dl-machine/blob/master/scripts/install-deeplearning-libraries.sh
-
-# TODO
-# Make this more parametric.
 
 ###################################
 #   Ubuntu Install script for:
 # - Nvidia graphic drivers
-# - Cuda 9.0 or 9.1
+# - Cuda 10.0
 # - cuDNN 7
 ###################################
 
@@ -27,12 +22,13 @@ else
     echo "Unsopported OS release"
     exit 2
 fi
-MACHINE = `uname -m` 
+MACHINE = `uname -m`
+ARCH = 'uname -p'
 
 export DEBIAN_FRONTEND=noninteractive
 
-apt-get update
-apt-get install -y git wget linux-image-generic build-essential unzip
+apt update
+apt install -y git wget linux-image-generic build-essential unzip
 
 # nvidia graphics drivers 384
 # see http://docs.nvidia.com/deeplearning/sdk/cudnn-install/                                       
@@ -53,11 +49,7 @@ apt -y install nvidia-384
 
 export ROOT_URL=http://developer.download.nvidia.com/compute/cuda/repos
 
-# Cuda 9.0 (TensorFlow 1.5 requires this)
-# export CUDA_VERSION=9.0.176-1
-# Cuda 9.1
-# export CUDA_VERSION=9.1.85-1
-# Cuda 9.2
+# Cuda 10.0
 export CUDA_VERSION=9.2.148-1
 
 # see: http://docs.nvidia.com/cuda/cuda-installation-guide-linux/