Upgrade system:
sudo apt-get update
sudo apt-get upgrade
sudo apt install nvidia-415*
sudo reboot
- Download miniconda from the oficial website. (Recommended: Python 3.* , 64-bits)
- Install miniconda. (Recommended: use predefined paths and answer yes whenever you are asked yes/no)
bash ~/Downloads/Miniconda3-latest-Linux-x86_64.sh
- Create and activate conda environment for the project
conda create -n ClothShape source activate ClothShape
- Install Pytorch
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
- Install other rand dependencies
conda install matplotlib opencv pillow scikit-learn scikit-image cython tqdm pip install tensorboardX
- Deactivate environment
conda deactivate
- Create and activate conda environment for tensorboard:
conda create -n tensorboard python=3.6 source activate tensorboard
- Install Tensorflow CPU
pip install tensorflow
- Deactivate environment
conda deactivate
- Run
python src/train.py --exp_dir experiments/model1/basic_settings
- To visualize. In a new terminal run:
source activate tensorboard tensorboard --logdir path/to/repo/experiments/model1/basic_settings/
- Run
results will be store in
python src/test.py --exp_dir experiments/model1/basic_settings
experiments/model1/basic_settings/test