19 lines
460 B
Bash
Executable file
19 lines
460 B
Bash
Executable file
#! /bin/bash
|
|
|
|
# Install dependecies
|
|
# 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
|
|
|
|
# Compile CUDA kernel for CD/EMD loss
|
|
root=`pwd`
|
|
cd metrics/pytorch_structural_losses/
|
|
make clean
|
|
make
|
|
cd $root
|