Neural Network Libraries v1.37.0とnnabla-rl v0.14.0、nnabla-nas v0.15.0をリリースしました！

2023年11月22日水曜日

Neural Network Libraries v1.37.0 をリリースしました！
また、nnabla-rl v0.14.0 と nnabla-nas v0.15.0 もリリースしました！
重要なリリースアイテムはSpotlightをご確認ください。

Spotlight

nnabla: Support CUDA 12.0.1

このバージョンからCUDA 12.0のサポートを開始します。
CUDA 12.0をご利用になるには以下のように nnabla-ext-cuda120 を指定してインストールしてください。

pip install nnabla-ext-cuda120

PCにCUDAの環境をセットアップされていない場合は、all-in-one wheelを利用する方法とDocker Imageを利用する方法があります。
どちらもnvidia-driverがインストールされている必要はありますが、CUDA 12.0 runtimeを同梱しているため、お手軽にCUDAをご利用いただくことができます。

pip install https://nnabla.org/whl/nnabla_ext_cuda_alllib/nnabla_ext_cuda_alllib-1.37.0-cp310-cp310-manylinux_2_17_x86_64.whl

Dockerをご利用になる場合は以下のようになります：

docker pull nnabla/nnabla-ext-cuda-multi-gpu:py310-cuda120-mpi4.1.3-v1.37.0
docker run --rm -it --gpus all nnabla/nnabla-ext-cuda-multi-gpu:py310-cuda120-mpi4.1.3-v1.37.0

上記以外のタグは DockerHub(nnabla-ext-cuda-multi-gpu) や DockerHub(nnabla-ext-cuda) を参照してください。

インストールするnvidia driverですが、CUDA 12.0をご利用になる際はLinuxではR525.85.12以降、WindowsではR528.33以降のnvidia driverが必要になります。
また、NVIDIAのData Center GPUをご利用の場合は、R470以降のドライバがホスト環境にインストールされていれば、cuda-compat-12-0 を利用し、環境変数を適切に設定することで、CUDA12.0をご利用になれます。（上記Docker Imageにはcuda compatが既に含まれています）
詳しくはNVIDIAのForward Compatibilityのページをご確認ください。

sudo mkdir /usr/local/share/keyrings
$ curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub  | sudo gpg --dearmor -o /usr/local/share/keyrings/nvidia-cuda-keyring.gpg
$ sudo echo "deb [signed-by=/usr/local/share/keyrings/nvidia-cuda-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64 /" > /etc/apt/sources.list.d/nvidia-cuda.list
apt-get update
apt-get install cuda-compat-12-0

nnabla-rl: Add sample rl project template

強化学習のプロジェクトを立ち上げたい、けれど何をどのように書けば学習ができるのかわからないといった方向けに、nnabla-rlを使って強化学習プロジェクトを開始するためのテンプレートを用意しました!
テンプレートを手元にコピーして、コードを見ながら目的に合わせてどう修正すれば簡単にわかるようになっています。
是非活用ください！