概要
数値演算や行列演算をPythonのNumPy/TensorFlowみたいに実行できる、Elixir__製OSSライブラリ__Nx（Numerical Elixir）をさわるためのDocker実行環境です。
- IExで実行します
実行環境
- macOS ローカル（M1でない）
- Docker version 20.10.5
セットアップ（準備）
tree
(任意のディレクトリ)
├── app
│ └── Dockerfile
└── docker-compose.yml
上記のディレクトリ構成で、各Dockerファイルを準備します
app/Dockerfile
FROM elixir:1.12.1-slim
WORKDIR /usr/src/app
RUN apt-get update -y && \
apt-get upgrade -y && \
apt-get install -y \
build-essential \
curl \
git \
gzip \
inotify-tools \
ssh \
sudo \
tar
RUN mix local.hex --force && \
mix local.rebar --force
RUN apt-get clean
docker-compose.yml
version: "3"
services:
app:
build: ./app
volumes:
- ./app:/usr/src/app
stdin_open: true
tty: true
セットアップ（Elixir）
terminal
$ docker-compose build
terminal
$ docker-compose up -d
Elixir1.12/OTP24が入ったコンテナが起動できるか確認
terminal
$ docker-compose exec app iex
Erlang/OTP 24 [erts-12.0.2] [source] [64-bit] [smp:4:4] [ds:4:4:10] [async-threads:1] [jit]
Interactive Elixir (1.12.1) - press Ctrl+C to exit (type h() ENTER for help)
iex(1)>
セットアップ（Nx）
ルート直下で、（例）「my_app」というプロジェクト名で mix new します
terminal
$ docker-compose run --rm app mix new . --app my_app
Creating nx_sandbox_app_run ... done
* creating README.md
* creating .formatter.exs
* creating .gitignore
* creating mix.exs
* creating lib
* creating lib/my_app.ex
* creating test
* creating test/test_helper.exs
* creating test/my_app_test.exs
Your Mix project was created successfully.
You can use "mix" to compile it, test it, and more:
mix test
Run "mix help" for more commands.
生成されたmix.exsに以下を追加
app/mix.exs
defp deps do
[
{:nx, "~> 0.1.0-dev", github: "elixir-nx/nx", branch: "main", sparse: "nx"} # --> add
...
mix deps.get実行
terminal
$ docker-compose run --rm app mix deps.get
mix test実行
terminal
$ docker-compose run --rm app mix test
Creating nx_sandbox_app_run ... done
==> nx
Compiling 20 files (.ex)
Generated nx app
==> my_app
Compiling 1 file (.ex)
Generated my_app app
..
Finished in 0.06 seconds (0.00s async, 0.06s sync)
1 doctest, 1 test, 0 failures
Randomized with seed 559301
ここまでで準備は完了です
使い方
terminal
$ docker-compose up -d
terminal
$ docker-compose exec app iex -S mix
Erlang/OTP 24 [erts-12.0.2] [source] [64-bit] [smp:4:4] [ds:4:4:10] [async-threads:1] [jit]
==> nx
Compiling 20 files (.ex)
Generated nx app
==> my_app
Compiling 1 file (.ex)
Generated my_app app
Interactive Elixir (1.12.1) - press Ctrl+C to exit (type h() ENTER for help)
iex(1)>t = Nx.tensor([[1, 2], [3, 4]])
#Nx.Tensor<
s64[2][2]
[
[1, 2],
[3, 4]
]
>
iex(2)> Nx.shape(t)
{2, 2}
>
iex(3)> Nx.divide(Nx.exp(t), Nx.sum(Nx.exp(t)))
#Nx.Tensor<
f32[2][2]
[
[0.032058604061603546, 0.08714432269334793],
[0.23688282072544098, 0.6439142227172852]
]
>
コンテナを立ち上げて、iex -S mixを起動して、IEx上でNxを実行することができるようになりました
（IEx停止は「Ctrl + c」x 2回）
参考
-
GitHub (Nx)
-
Docker Image (Elixir)