# # conda installation
# wget https://repo.anaconda.com/miniconda/Miniconda3-py38_4.10.3-Linux-x86_64.sh
# chmod +x Miniconda3-py38_4.10.3-Linux-x86_64.sh
# bash Miniconda3-py38_4.10.3-Linux-x86_64.sh -b -f -p /home/user/.local
# rm Miniconda3-py38_4.10.3-Linux-x86_64.sh
# conda config --env --set always_yes true
# conda update -n base -c defaults conda -y

# # conda environment setup
# conda env create -f environment.yaml
# conda init bash
# source /home/user/.bashrc
# source activate icon
nvidia-smi
pip install torch==1.8.2 torchvision==0.9.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu111
pip install -r requirement.txt