Google Colabでfastai
最先端の深層(機械)学習を実務家が気軽に適用できるようにするための仕組みとしてfastaiがある.「AIをもう一度uncoolに」を標語にしているように,専門家でなくても(Pythonを知っていれば)ある程度(というか数年前の世界新記録程度)の深層学習を使うことができる.
問題はインストールが面倒なことだ.GPUつきのlinuxが必要で,それ上にPyTorchを入れた後でインストールする必要がある.WindowsやMacは今のところサポートしておらず,GPUの種類にあわせて半日かけてドライバ,Cudaを入れた後にトライしてもバグで動かないということも多々ある(そのうち直ると思うがversion 1.xは出たばかりなので不安定だ).
(今のところ)無料のGoogle Co laboratoryでインストールできたので,これをお奨めする.
1.メニューのランタイム/ランタイムのタイプを変更でGPUありにする.
2.以下を入れて,Shift Enter
!pip3 install https://download.pytorch.org/whl/cu80/torch-1.0.0-cp36-cp36m-linux_x86_64.whl
!pip3 install torchvision
!pip3 install fastai
!pip3 install Pillow==4.0.0
とまあこれだけだ.
追記:その後,fastaiの新しい講義が始まり,インストーラーも公開された.
!curl https://course-v3.fast.ai/setup/colab | bash
で動くようになったようだ.
インストールした環境をチェックしておく.
from fastai.utils.collect_env import *
show_install(True)
python : 3.6.7
fastai : 1.0.39
fastprogress : 0.1.18
torch : 1.0.0
nvidia driver : 396.44
torch cuda : 8.0.61 / is available
torch cudnn : 7102 / is enabled
=== Hardware ===
nvidia gpus : 1
torch devices : 1
- gpu0 : 11441MB | Tesla K80
=== Environment ===
platform : Linux-4.14.65+-x86_64-with-Ubuntu-18.04-bionic
distro : #1 SMP Sun Sep 9 02:18:33 PDT 2018
conda env : Unknown
python : /usr/bin/python3
sys.path :
/env/python
/usr/lib/python36.zip
/usr/lib/python3.6
/usr/lib/python3.6/lib-dynload
/usr/local/lib/python3.6/dist-packages
/usr/lib/python3/dist-packages
/usr/local/lib/python3.6/dist-packages/IPython/extensions
/root/.ipython
-----------------------------------------------------------------------------+
| NVIDIA-SMI 396.44 Driver Version: 396.44 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 Tesla K80 Off | 00000000:00:04.0 Off | 0 |
| N/A 27C P8 27W / 149W | 11MiB / 11441MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| No running processes found |
+-----------------------------------------------------------------------------+
この記事が気に入ったらサポートをしてみませんか?