Installation.md

May 10, 2023 ยท View on GitHub

Installation

Our codebase is developed based on Ubuntu 18.04 and Pytorch framework.

Requirements

  • Python >=3.8
  • Pytorch >=1.7.1
  • torchvision >= 0.8.2
  • CUDA >= 10.1
  • cuDNN (if CUDA available)

Installation with conda

# We suggest to create a new conda environment with python version 3.8
conda create --name PHMR python=3.8

# Activate conda environment
conda activate PHMR

# Install Pytorch that is compatible with your CUDA version
# CUDA 10.1
conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=10.1 -c pytorch
# CUDA 10.2
conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=10.2 -c pytorch
# CUDA 11.1
conda install pytorch==1.8.0 torchvision==0.9.0 cudatoolkit=11.1 -c pytorch -c conda-forge

# Install apex
git clone https://github.com/NVIDIA/apex.git
cd apex
python setup.py install --cuda_ext --cpp_ext
cd ..

# Install OpenDR
pip install matplotlib
pip install git+https://gitlab.eecs.umich.edu/ngv-python-modules/opendr.git

# Install PointHMR
git clone --recursive https://github.com/DCVL-3D/PointHMR_release
cd PointHMR_release
python setup.py build develop

# Install requirements
pip install -r requirements.txt
pip install ./manopth/.
pip install --upgrade azureml-core