Skip to content

GitLab

Commit 8b2ba9ca authored by Jeffrey Morgan's avatar Jeffrey Morgan
Browse files

minor improvements to `install.sh`

parent e29662ab
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
scripts/install.sh
View file @ 8b2ba9ca
...@@ -125,17 +125,18 @@ install_cuda_driver_yum() { ...@@ -125,17 +125,18 @@ install_cuda_driver_yum() {
rhel) rhel)
status 'Installing EPEL repository...' status 'Installing EPEL repository...'
# EPEL is required for third-party dependencies such as dkms and libvdpau # EPEL is required for third-party dependencies such as dkms and libvdpau
$SUDO $PACKAGE_MANAGER -y install https://dl.fedoraproject.org/pub/epel/epel-release-latest-$2.noarch.rpm $SUDO $PACKAGE_MANAGER -y install https://dl.fedoraproject.org/pub/epel/epel-release-latest-$2.noarch.rpm || true
;; ;;
esac esac
status 'Installing CUDA driver...' status 'Installing CUDA driver...'
$SUDO $PACKAGE_MANAGER -y update $SUDO $PACKAGE_MANAGER -y update
$SUDO $PACKAGE_MANAGER -y install cuda-drivers
if [ "$1" = 'centos' ] || [ "$1$2" = 'rhel7' ]; then if [ "$1" = 'centos' ] || [ "$1$2" = 'rhel7' ]; then
$SUDO $PACKAGE_MANAGER -y install nvidia-driver-latest-dkms $SUDO $PACKAGE_MANAGER -y install nvidia-driver-latest-dkms
fi fi
$SUDO $PACKAGE_MANAGER -y install cuda-drivers
} }
# ref: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#ubuntu # ref: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#ubuntu
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment