Neural Network Libraries v1.34.0をリリースしました！
重要なリリースアイテムについてはSpotlightをご覧ください。
Remarks
重大なバグ修正
過去のバージョンにおいて、SyncBatchNormalizationCudaのchannel lastレイアウトでの実行時のvarianceの計算で不具合が発生していたため、それを修正しました。詳細についてはプルリクエストをご参照ください。
Spotlight
CUDA11.6 Support
このバージョンからCUDA 11.6のサポートを開始します。CUDA 11.4の提供は終わりますが、CUDAの互換性があるため、今までの環境はそのまま継続してご利用できます。
CUDA 11.6をご利用になるには以下のように nnabla-ext-cuda116 を指定してインストールしてください。
pip install nnabla-ext-cuda116
もしくは、まだPCにCUDAの環境をセットアップされていない場合は、all-in-one wheel を利用することもできます。
このwheelはCUDA 11.6 runtimeを同梱しているため、nvidia-driverだけをインストールすればCUDA extensionをご利用いただけます。
pip install https://nnabla.org/whl/nnabla_ext_cuda_alllib/nnabla_ext_cuda_alllib-1.34.0-cp310-cp310-manylinux_2_17_x86_64.whl
またDocker imageも各バージョンごとに用意していますので、こちらも併せてご利用ください。
docker pull nnabla/nnabla-ext-cuda-multi-gpu:py310-cuda116-mpi4.1.3-v1.34.0
詳しくは DockerHub(nnabla-ext-cuda-multi-gpu) や DockerHub(nnabla-ext-cuda) を参照してください。
nnabla
Bugfix
- Check consistency among memory layouts in SyncBatchNormalization
- Suppress unnecessary error msg for the deallocation of a child array
- Fix SyncBatchnormalizationCuda in channel last mode
- Add NonZero, NonMaxSuppression, OneHot, and Resize support to ONNX importer
Build
- CUDA11.6 Support (CPU / GPU)
- fix alllib wheel dependency in cuda11.6
- Sync api_level version from NNabla.
