Installation

Jump to bottom

grasseau edited this page Dec 20, 2019 · 9 revisions

Installation

git clone https://github.com/LLRCMS/MDT.git  
cd MDT  
. ./torch.env   
cd MDT   
virtualenv -p python3 venv   
source venv/bin/activate  
python3 -m pip install -e .  
python3 -m pip install opencv-python

Kernel compilation for V100

cd cuda_functions/nms_2D/src/cuda/  
nvcc -c -o nms_kernel.cu.o nms_kernel.cu -x cu -Xcompiler -fPIC -arch=sm_70  
cd ../..  
python build.py  
cd ../..  
cd cuda_functions/roi_align_2D/roi_align/src/cuda/  
nvcc -c -o crop_and_resize_kernel.cu.o crop_and_resize_kernel.cu -x cu -Xcompiler -fPIC -arch=sm_70  
cd ../..  
python build.py

The same with 3D kernels

cd cuda_functions/nms_3D/src/cuda/
...

Install Batchgenerators

cd ../batchgenerators
python3 -m pip install -e .