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