LoginSignup
34
33

More than 5 years have passed since last update.

Amazon EC2のGPUインスタンスにChainer v1.6を1行で入れる

Last updated at Posted at 2016-01-20

Chainer 1.7も1行で入ることを確認しました

v1.5では3行も必要だったのが、1行になりました。
例のごとく、nVidia公式の "Amazon Linux AMI with NVIDIA GRID GPU Driver" を使います。起動して、ログインしたら、以下のコマンドを実行して下さい。

$ sudo CUDA_PATH=/opt/nvidia/cuda pip install chainer

numpyのビルドなども走るので、結構時間はかかります。気長に待ちましょう。
CUDAが正しくサポートできているかは、import cupyできるかで確認できます。

解説

nVidia公式のAMIでは、CUDAは /opt/nvidia/cuda にあります。これを環境変数 CUDA_PATH でインストール時に渡してあげると、インストールできます。本当はnvccコマンドの場所から逆算するのですが、 sudoでインストールしようとするとPATH環境変数をクリア してしまうのでこうしています。

HDF5に対応する

HDF5のシリアライザはオプショナルになりました。あとからh5pyをインストールすれば使えるようになります。

$ sudo yum-config-manager --enable epel
$ sudo yum install -y hdf5-devel
$ sudo pip install h5py
34
33
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
34
33