iris-bench/setup_vm_tmux.sh at main · stfc/iris-bench · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#!/bin/bash

# =============================================================================
# setup_vm_tmux.sh
# =============================================================================
# This script sets up a virtual machine for running benchmarks and iris-gpubench
# with Tmux on Ubuntu. The Ubuntu VM should be equipped with Nvidia GPUs.
# It installs essential software packages for iris-gpubench including Git, Python,
# Tmux, and NVIDIA drivers.
# =============================================================================

#### INSTALL Git, python3-pip and Venv ###

echo -e "/n~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~/nInstall git, python3-pip, and Venv/n~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~/n"

# Install Git, Python, and venv
sudo apt-get install -y git
sudo apt-get install -y wget python3-pip
sudo apt-get install -y python3-venv

#### INSTALL TMUX


echo -e "/n~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~/nINSTALL Tmux/n~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~/n"

sudo apt update
sudo apt install tmux


##### INSTALL NVIDIA DRIVERS #####

echo -e "/n~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~/nINSTALL NVIDIA DRIVERS/n~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~/n"

# Install NVIDIA drivers (if needed)
# Ensure the driver is up-to-date for the GPU to achieve maximum performance.
sudo apt install -y nvidia-driver-535

##### REBOOT VM #####

echo -e "/n~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~/nREBOOT/n~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~/n"

# Reboot the system to finish setup
sudo reboot