#!/bin/bash

snap remove lxd
snap remove core18
apt -y purge snapd gdm3 colord accountsservice rsyslog switcheroo-control \
       network-manager multipath-tools modemmanager upower rtkit policykit-1
apt -y autoremove

apt update
apt -y install python3-pip python-is-python3 nvtop zstd rclone

pip3 install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 \
          -f https://download.pytorch.org/whl/torch_stable.html
pip3 install scikit-learn tqdm tensorboard

cd /root

git clone https://git.jordangong.com/jordangong/gait-recognition.git
cd gait-recognition; git checkout python3.8; cd ..
mkdir -p gait-recognition/data/CASIA-B-MRCNN
wget https://storage.googleapis.com/gait-dataset/CASIA-B-MRCNN-SEG.tar.zst
tar -I zstd -xf CASIA-B-MRCNN-SEG.tar.zst -C gait-recognition/data/CASIA-B-MRCNN
mv gait-recognition/data/CASIA-B-MRCNN/{CASIA-B-MRCNN-SEG,SEG}