build_windows.md

Build CUDA extension on Windows

Prerequisites

At first, clone nnabla and nnabla-ext-cuda into same folder.

Then, install CUDA11.6/cuDNN8.4 or CUDA12.0/cuDNN8.8 from following site.

• CUDA
• https://developer.nvidia.com/cuda-toolkit-archive

Get several versions of cuDNN from following site. (Registration required)

• cuDNN
• https://developer.nvidia.com/rdp/cudnn-archive

Before building nnabla-ext-cuda, build nnabla with Instruction.

Build cpplib

You can build cpplib for nnabla-ext-cuda with following command.

Note: parameters CUDA_VERSION and CUDNN_VERSION are options of cuda/cudnn versions.

build-tools\msvc\build_cpplib.bat CUDA_VERSION CUDNN_VERSION

For examples:

build-tools\msvc\build_cpplib.bat 11.6 8

or:

build-tools\msvc\build_cpplib.bat 12.0 8

Tested version

CUDA/cuDNN: 11.6/8.4 12.0/8.8

Build wheel

Note: parameters PYTHON_VERSION, CUDA_VERSION and CUDNN_VERSION are options of python, cuda and cudnn versions.

build-tools\msvc\build_wheel.bat PYTHON_VERSION CUDA_VERSION CUDNN_VERSION

For examples:

build-tools\msvc\build_wheel.bat 3.10 11.6 8

or:

build-tools\msvc\build_wheel.bat 3.10 12.0 8

Tested version

PYTHON: 3.9 3.10
CUDA/cuDNN: 11.6/8.4 12.0/8.8