JupyterNotebookで実行しているPythonプログラムにGPUが割り当てられない

GPUが割り当てられない問題

ipythonだったら

os.environ["CUDA_VISIBLE_DEVICES"]="0"

と書いておけば割り当てることが出来たのに…。

実際は以下のように割り当てられていない。

$ nvidia-smi
Sun Aug 26 18:20:39 2018       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 384.111                Driver Version: 384.111                   |
|-------------------------------+----------------------+----------------------+
| 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           On   | 00000000:00:1E.0 Off |                    0 |
| N/A   28C    P8    31W / 149W |      0MiB / 11439MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

解決策

マウントした先でプログラムを実行していてもう一度色々インストールしなくちゃいけないの忘れてただけでした。

$ conda install tensorflow-gpu