Released Neural Network Libraries v1.34.0! CUDA 11.6 support, etc.
We have released Neural Network Libraries v1.34.0!
Please see “Spotlight” section for important changes.
Remarks
Important bugfix
Fixed a bug in the variance calculation in channel last layout of SyncBatchNormalizationCuda in previous versions. See the pull request for details.
Spotlight
CUDA 11.6 Support
We started supporting CUDA 11.6 extension. And CUDA 11.4 extension is removed from this version, but users can keep using current CUDA environment because it has compatibility.
To use CUDA 11.6 extension, specify nnabla-ext-cuda116 and install it by:
pip install nnabla-ext-cuda116Otherwise, if you don’t setup CUDA environment on your computer, you can use all-in-one wheel. This wheel bundled with CUDA 11.6 runtime, so you only install nvidia-driver to use this wheel.
pip install https://nnabla.org/whl/nnabla_ext_cuda_alllib/nnabla_ext_cuda_alllib-1.34.0-cp310-cp310-manylinux_2_17_x86_64.whlIn addition, docker images are also available for each version, so please use these as well!
docker pull nnabla/nnabla-ext-cuda-multi-gpu:py310-cuda116-mpi4.1.3-v1.34.0Please refer DockerHub(nnabla-ext-cuda-multi-gpu) or DockerHub(nnabla-ext-cuda) for detail.