2019-07-14 04:32:26 +00:00
|
|
|
#! /bin/bash
|
|
|
|
|
|
|
|
# Install dependecies
|
2022-04-07 14:05:49 +00:00
|
|
|
# conda install pytorch==1.0.1 torchvision==0.2.2 cudatoolkit=10.0 -c pytorch
|
|
|
|
# conda install matplotlib tqdm scikit-learn -y
|
|
|
|
# pip install pillow==5.0.0
|
|
|
|
# pip install scipy==1.0.1
|
|
|
|
# pip install numpy==1.16.4
|
|
|
|
# pip install tensorflow-gpu==1.13.1
|
|
|
|
# pip install tensorboardX==1.7
|
|
|
|
# pip install torchdiffeq==0.0.1
|
2019-07-14 04:32:26 +00:00
|
|
|
|
|
|
|
# Compile CUDA kernel for CD/EMD loss
|
2022-04-07 14:05:49 +00:00
|
|
|
root=`pwd`
|
2019-07-14 04:32:26 +00:00
|
|
|
cd metrics/pytorch_structural_losses/
|
|
|
|
make clean
|
|
|
|
make
|
|
|
|
cd $root
|