wehub-resource-sync
21 lines
764 B
Bash
Executable File
21 lines
764 B
Bash
Executable File
#! /bin/bash
|
|
|
|
# This script updates the llmware 0.1.3 AMI with Cuda and required dependencies to take advantage of Nvidia GPU instances
|
|
|
|
# Setup cuda keyring so that we just have to run a single 'apt update' below
|
|
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
|
|
sudo dpkg -i cuda-keyring_1.1-1_all.deb
|
|
rm cuda-keyring_1.1-1_all.deb
|
|
|
|
# Install ubuntu + nvidia drivers
|
|
sudo apt update
|
|
sudo apt install -y ubuntu-drivers-common nvidia-driver-535
|
|
|
|
# Install cuda
|
|
sudo apt install -y cuda
|
|
|
|
# Update .bashrc and the current shell's environment
|
|
echo "export PATH=/usr/local/cuda/bin:${PATH}" >> ~/.bashrc
|
|
echo "export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64:${LD_LIBRARY_PATH}" >> ~/.bashrc
|
|
source ~/.bashrc
|