ArrayFireと言う行列計算などを高速化できるライブラリをPythonのラッパーから使用して、MNISTのサンプルを試してみました。
結構CUDA Toolkit関連のセットアップに時間を取られましたが、上手くいきました。使用した環境はUbuntu 20.04でCUDA Toolkitは11.4です。
これが動画です。
I used ArrayFire, a library that enables fast matrix calculations etc using a Python wrapper. I tested a MNIST sample. Setting up CUDA Toolkit and other things took my time, but it went well. The environment I used was Ubuntu 20.04 and CUDA Toolkit 11.4.
Below is the video of it running.
これが動画です。
I used ArrayFire, a library that enables fast matrix calculations etc using a Python wrapper. I tested a MNIST sample. Setting up CUDA Toolkit and other things took my time, but it went well. The environment I used was Ubuntu 20.04 and CUDA Toolkit 11.4.
Below is the video of it running.
コメント
コメントを投稿