Skip to content

Latest commit

 

History

History
46 lines (32 loc) · 887 Bytes

INSTALL.md

File metadata and controls

46 lines (32 loc) · 887 Bytes

Installation

Our codebase is developed based on NVIDIA GPU cards (A100 40GB).

Requirements

  • Python 3.8
  • Pytorch 1.10
  • torchvision 0.11.0
  • cuda 11.3

Setup with Conda

We suggest to create a new conda environment and install all the relevant dependencies.

# Create a new environment
conda create --name deformer python=3.8
conda activate deformer

# Install Pytorch
conda install pytorch==1.10.0 torchvision==0.11.0 cudatoolkit=11.3 -c pytorch -c conda-forge

export INSTALL_DIR=$PWD

# Install DeFormer
cd $INSTALL_DIR
git clone https://github.com/yusukey03012/DeFormer
cd DeFormer
python setup.py build develop

# Install requirements
pip install -r requirements.txt

# Install Deepspeed
pip install deepspeed

# Compiling CUDA operators for deformable Detr
cd ./src/modeling/core/models/ops
sh ./make.sh

cd $INSTALL_DIR
cd DeFormer
unset INSTALL_DIR