add requirements for gpu inference

sct-pipeline / contrast-agnostic-softseg-spinalcord

Contrast-agnostic spinal cord segmentation project with softseg

MIT License

4 stars 3 forks source link

Closed naga-karthik closed 9 months ago

naga-karthik commented 9 months ago

Current inference requirements only support CPU inference. This PR adds the requirements for GPU inference (for advanced users)

Fixes #75

naga-karthik commented 9 months ago

Thanks @rohanbanerjee and @sandrinebedard! Merging!