baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

minor improvements to install.sh

Browse source
This commit is contained in:
Jeffrey Morgan 2023-09-23 11:20:39 -04:00
parent e29662ab5c
commit 8b2ba9cab8
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
scripts/install.sh
View file

@ -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