PULP-TrainLib-Tutorial/install_ub22.sh at main · dnadalini/PULP-TrainLib-Tutorial · 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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
#!/bin/bash
echo "Beginning setup.."
# General setup
sudo apt-get update
sudo apt-get install -y make python-is-python3

# Setup repo and environment
git submodule update --init --recursive
conda create --name trainlib-tutorial
conda activate trainlib-tutorial

# Download dependencies for pulp-sdk
sudo apt-get install -y build-essential git libftdi-dev libftdi1 doxygen python3-pip libsdl2-dev curl cmake libusb-1.0-0-dev scons gtkwave libsndfile1-dev rsync autoconf automake texinfo libtool pkg-config libsdl2-ttf-dev
pip install --user argcomplete pyelftools

# Download gnu-gcc-toolchain
wget https://github.com/pulp-platform/pulp-riscv-gnu-toolchain/releases/download/v1.0.16/v1.0.16-pulp-riscv-gcc-ubuntu-18.tar.bz2
tar -xf v1.0.16-pulp-riscv-gcc-ubuntu-18.tar.bz2 -C ./
mv v1.0.16-pulp-riscv-gcc-ubuntu-18/ pulp-riscv-gcc-toolchain/
rm -rf v1.0.16-pulp-riscv-gcc-ubuntu-18.tar.bz2

# Set up GCC version 9.4.0
sudo apt-get update -y && \
sudo apt-get upgrade -y && \
sudo apt-get dist-upgrade -y && \
sudo apt-get install build-essential software-properties-common -y && \
sudo add-apt-repository ppa:ubuntu-toolchain-r/test -y && \
sudo apt-get update -y && \
sudo apt-get install gcc-9 g++-9 -y && \
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 60 --slave /usr/bin/g++ g++ /usr/bin/g++-9 && \
sudo update-alternatives --config gcc

# Build pulp-sdk
cd pulp-riscv-gcc-toolchain
source sourceme.sh
cd ..
USR_PATH=$(pwd)/pulp-riscv-gcc-toolchain
export PULP_RISCV_GCC_TOOLCHAIN=$USR_PATH
export PATH=$USR_PATH/bin:$PATH
cd pulp-sdk/
source configs/pulp-open.sh
make build
cd ..

# Download dependencies for pulp-trainlib
python -m pip install argparse
#python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
python -m pip install install torch torchvision torchaudio
python -m pip install torchsummary

# Checkout the correct version of PULP-TrainLib
cd pulp-trainlib
git pull origin
git checkout trainlib-tutorial
cd ..

# Setup GVSoC's memory for this tutorial (set L1 to 256 kB)
rm pulp-sdk/rtos/pulpos/pulp/kernel/chips/pulp/link.ld
cp img/link.ld pulp-sdk/rtos/pulpos/pulp/kernel/chips/pulp/
rm pulp-sdk/tools/gap-configs/configs/chips/pulp/pulp.json
cp img/pulp.json pulp-sdk/tools/gap-configs/configs/chips/pulp/

echo "Setup successful!"