Skip to content

Latest commit

 

History

History
53 lines (42 loc) · 1.26 KB

INSTALL.md

File metadata and controls

53 lines (42 loc) · 1.26 KB
Raw

Installation

This repository is built in PyTorch 1.8.1 and tested on Ubuntu 16.04 environment (Python3.7, CUDA10.2, cuDNN7.6). Follow these intructions

  1. Clone our repository
git clone https://github.com/swz30/Restormer.git
cd Restormer
  1. Make conda environment
conda create -n pytorch181 python=3.7
conda activate pytorch181
  1. Install dependencies
conda install pytorch=1.8 torchvision cudatoolkit=10.2 -c pytorch
pip install matplotlib scikit-learn scikit-image opencv-python yacs joblib natsort h5py tqdm
pip install einops gdown addict future lmdb numpy pyyaml requests scipy tb-nightly yapf lpips
  1. Install basicsr
python setup.py develop --no_cuda_ext

Download datasets from Google Drive

To be able to download datasets automatically you would need go and gdrive installed.

  1. You can install go with the following
curl -O https://storage.googleapis.com/golang/go1.11.1.linux-amd64.tar.gz
mkdir -p ~/installed
tar -C ~/installed -xzf go1.11.1.linux-amd64.tar.gz
mkdir -p ~/go
  1. Add the lines in ~/.bashrc
export GOPATH=$HOME/go
export PATH=$PATH:$HOME/go/bin:$HOME/installed/go/bin
  1. Install gdrive using
go get github.com/prasmussen/gdrive
  1. Close current terminal and open a new terminal.