conda install -c conda-forge opencv
conda install numpy scipy matplotlib
conda install -c conda-forge pbzip2 pydrive
conda install pillow tqdm
pip install GPUtil
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch