-
Notifications
You must be signed in to change notification settings - Fork 23
/
install.sh
100 lines (74 loc) · 2.61 KB
/
install.sh
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
#!/bin/bash
if [ "$#" -ne 2 ]; then
echo "ERROR! Illegal number of parameters. Usage: bash install.sh conda_install_path environment_name"
exit 0
fi
conda_install_path=$1
conda_env_name=$2
source $conda_install_path/etc/profile.d/conda.sh
echo "****************** Creating conda environment ${conda_env_name} python=3.8.5 ******************"
conda create -y --name $conda_env_name python=3.8.5
echo ""
echo ""
echo "****************** Activating conda environment ${conda_env_name} ******************"
conda activate $conda_env_name
echo ""
echo ""
echo "****************** Installing pytorch with cuda10.2 ******************"
conda install -y pytorch torchvision cudatoolkit=10.2 -c pytorch
echo ""
echo ""
echo "****************** Installing matplotlib ******************"
conda install -y matplotlib
echo ""
echo ""
echo "****************** Installing opencv, scipy, tqdm, exifread ******************"
pip install opencv-python
pip install scipy
pip install tqdm
pip install exifread
echo ""
echo ""
echo "****************** Installing cupy (dependency for pwcnet) ******************"
conda install -y -c conda-forge cupy=7.8.0
echo ""
echo ""
echo "****************** Installing rawpy ******************"
pip install rawpy
echo ""
echo ""
echo "****************** Installing LPIPS ******************"
pip install lpips
echo ""
echo ""
echo "****************** Installing tensorboard ******************"
pip install tb-nightly
echo ""
echo ""
echo "****************** Installing jpeg4py python wrapper ******************"
pip install jpeg4py
echo ""
echo ""
echo "****************** Installing gdown ******************"
pip install gdown
echo ""
echo ""
echo "****************** Setting up environment ******************"
python -c "from admin.environment import create_default_local_file; create_default_local_file()"
echo ""
echo ""
echo "****************** Downloading networks ******************"
mkdir pretrained_networks
echo ""
echo ""
echo "****************** Downloading PWCNet model ******************"
gdown https://drive.google.com/uc\?id\=1s11Ud1UMipk2AbZZAypLPRpnXOS9Y1KO -O pretrained_networks/pwcnet-network-default.pth
echo ""
echo "****************** Downloading DBSR synthetic model ******************"
gdown https://drive.google.com/uc\?id\=1bdtz_gr_m9MnypVqDigH6H1xoae82lwY -O pretrained_networks/dbsr_synthetic_default.pth
echo ""
echo "****************** Downloading DBSR burstsr model ******************"
gdown https://drive.google.com/uc\?id\=1GyfJzqBx9st9Qdvy2fRizWvxCbNOerFe -O pretrained_networks/dbsr_burstsr_default.pth
echo ""
echo ""
echo "****************** Installation complete! ******************"