some fixes

This commit is contained in:
Nicolas 2022-03-17 16:03:07 +01:00
parent 0414759c74
commit 4cd09db8a0
4 changed files with 29 additions and 29 deletions

View File

@ -1,4 +1,4 @@
# Setup for setup Tensorflow with GPU
# Setup for Tensorflow with GPU
Tested for ubuntu-20.04.4

View File

@ -14,7 +14,7 @@ if [ ! -s "~/.ssh/authorized_keys" ]; then
cat id_*.pub >> ~/.ssh/authorized_keys
fi
echo "\n"
echo ""
echo "Install and config python & jupyter..."
apt-get update && \
apt-get install -y ipmitool && modprobe ipmi_devintf && \
@ -34,7 +34,7 @@ if [ $? -eq 0 ]; then
echo "check nvidia drivers..."
nvidia-smi
echo "\n"
echo ""
echo "Setup successfully finished"
fi

View File

@ -18,24 +18,24 @@ while [ -b /dev/bcache0 ]
exit
fi
echo "Setup bcache... "
apt-get update && \
apt-get install -y bcache-tools && \
echo "Clean disks and setup bcache... "
dd if=/dev/zero if=$ssddev bs=512 count=8 && \
dd if=/dev/zero if=$hdddev bs=512 count=8 && \
wipefs -a $ssddev && \
wipefs -a $hdddev && \
make-bcache -C $ssddev -B $hdddev --writeback && \
echo $sequential_cutoff > /sys/block/bcache0/bcache/sequential_cutoff && \
echo "ACTION==\"add\", SUBSYSTEM==\"block\", ENV{MAJOR}==\"252\", ATTR{bcache/sequential_cutoff}=\"$sequential_cutoff\"" \
> /etc/udev/rules.d/99-bcache_sequential_cutoff.rules && \
udevadm test /sys/block/bcache0
apt-get update && \
apt-get install -y bcache-tools && \
make-bcache -C $ssddev -B $hdddev --writeback
if [ $? -eq 0 ]; then
echo "Wait until device is ready... "
while [ ! -b /dev/bcache0 ]; do sleep 1; done
echo "Fromat and mount bcache device... "
if [ $? -eq 0 ]; then
echo "Config, format and mount bcache device... "
echo $sequential_cutoff > /sys/block/bcache0/bcache/sequential_cutoff && \
echo "ACTION==\"add\", SUBSYSTEM==\"block\", ENV{MAJOR}==\"252\", ATTR{bcache/sequential_cutoff}=\"$sequential_cutoff\"" \
> /etc/udev/rules.d/99-bcache_sequential_cutoff.rules && \
udevadm test /sys/block/bcache0 && \
mkfs.ext4 /dev/bcache0 && \
mkdir /mnt/bcache && \
mount /dev/bcache0 /mnt/bcache && \
@ -43,7 +43,7 @@ if [ $? -eq 0 ]; then
fi
if [ $? -eq 0 ]; then
echo "Config home directory for new users on bcache partition... "
echo "Config default home directory for new users on bcache partition... "
mkdir /mnt/bcache/home && \
echo "#" >> /etc/default/useradd && \
echo "# Modifications:" >> /etc/default/useradd && \

View File

@ -23,14 +23,14 @@ fi
distribution=$(. /etc/os-release;echo $ID$VERSION_ID | sed -e 's/\.//g')
architecture="x86_64"
echo "\n"
echo "Install drivers..."
echo ""
echo "Install requirements..."
apt-get update && \
apt-get install -y linux-headers-$(uname -r) software-properties-common
if [ $? -eq 0 ]; then
echo "\n"
echo "Install the CUDA repository public GPG key..."
echo ""
echo "Setup the CUDA repository public GPG key..."
apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/$distribution/$architecture/7fa2af80.pub && \
wget https://developer.download.nvidia.com/compute/cuda/repos/$distribution/$architecture/cuda-$distribution.pin && \
mv cuda-$distribution.pin /etc/apt/preferences.d/cuda-repository-pin-600
@ -38,24 +38,26 @@ if [ $? -eq 0 ]; then
fi
if [ $? -eq 0 ]; then
echo "\n"
echo ""
echo "Setup the CUDA network repository..."
add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/$distribution/$architecture/ /"
fi
if [ $? -eq 0 ]; then
echo "\n"
echo ""
echo "install CUDA..."
#Installs all CUDA Toolkit and Driver packages. Handles upgrading to the next version of the cuda package when it's released.
#Use the --no-install-recommends option for a lean driver install without any dependencies on X packages
#Installs CUDA Toolkit and Driver packages.
#Use the --no-install-recommends option should prevent install any dependencies on X packages
#It still installes alot of dependencies, but fortunately it seams not to setup x-server
apt-get update && \
apt-get install -y --no-install-recommends cuda-11-2
fi
if [ $? -eq 0 ]; then
echo "\n"
echo ""
echo "install libcudnn8..."
#Latest available version from nvidia ubuntu1804 repo:
#Latest available version from nvidia ubuntu1804 repo
#For the current Ubuntu version it was not available. But it seams to work anyway.
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb && \
apt-get install -y ./libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb && \
rm ./libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb && \
@ -65,9 +67,7 @@ if [ $? -eq 0 ]; then
fi
if [ $? -eq 0 ]; then
#reboot system
#systemctl reboot
echo ""
echo "Setup successfully finished"
echo "Type \"systemctl reboot\" to reboot system"
echo "Type \"sudo systemctl reboot\" to reboot system for loading GPU drivers)"
fi