2019ǯ08·î29Æü
NVIDIA GPU¤Î¥É¥é¥¤¥Ð¥Ð¡¼¥¸¥ç¥ó¤ÈDocker Image¤ÎÃæ¤ÎCUDA¥Ð¡¼¥¸¥ç¥ó¤Î´Ø·¸¤Ë¤Ä¤¤¤Æ
Ubuntu ´Ä¶¤Ç¤ÎÈ÷˺Ͽ.
NVIDIA ¤Î GPU ¤òÀѤó¤Ç¤¤¤ë¤È¥É¥é¥¤¥Ð¤Î¥Ð¡¼¥¸¥ç¥ó¤È CUDA ¤Î¥Ð¡¼¥¸¥ç¥ó¤¬Â·¤ï¤ºÆñµ·¤¹¤ë¤³¤È¤¬Â¿¤¤.
ÆÃ¤Ë Docker ¥¤¥á¡¼¥¸¤Ë²¿¤ò»È¤¨¤Ð...¤ß¤¿¤¤¤ÊÏäˤʤë.
º£²ó¤Ï¤½¤ó¤ÊÏÃ.
¥Ï¥Þ¤Ã¤¿¤Î¤Ï TensorFlow Docker Image ¤ò»È¤ª¤¦¤È¤·¤¿»þ¤Ç¤·¤¿.
¤³¤³¤Ë¤¢¤ë Docker Image ¤ò GPU ¥¿¥°ÉÕ¤¤ÇÆ°¤«¤·¤¿¤È¤³¤í¤É¤¦¤ä¤é GPU ¤¬¸«¤¨¤Æ¤¤¤Ê¤¤¤È¤¤¤¦¤³¤È¤Ëµ¤¤¬¤Ä¤¤¤Æ¤·¤Þ¤¤¤Þ¤·¤¿.
¤È¤¤¤¦¤ï¤±¤Ç tensorflow/tensorflow ¤ò³Îǧ¤¹¤ë¤È...
¤È¤¤¤¦µ½Ò¤Ë.
¤Ä¤Þ¤ê CUDA ¤Î¥Ð¡¼¥¸¥ç¥ó¤Ï 10.0 ¤Ç¥¤¥á¡¼¥¸¤¬¥Ó¥ë¥É¤µ¤ì¤Æ¤¤¤¿¤è¤¦¤Ç¤¹.
¤³¤Î»þ, ¥Þ¥·¥ó¤Î¥Û¥¹¥È¤Î NVRM version ¤Ï 395 ¤Ç, »È¤¨¤ë CUDA version ¤Ï 9.2.
¤Ê¤Î¤Ç Dockerfile ¤Î CUDA ¤ò 9.2 ¤Ç¥Ó¥ë¥É¤¹¤ë¤«, GPU ¤Î¥É¥é¥¤¥Ð¤ò 10.0 ¤¬»È¤¨¤ë¤â¤Î¤Ë¥Ð¡¼¥¸¥ç¥ó¥¢¥Ã¥×¤¹¤ë¤«¤ÎÆóÂò¤Ç¤·¤¿.
Dockerfile ¤Î¥ê¥Ó¥ë¥É¤ÏÌÌÅݤÀ¤Ã¤¿¤Î¤Ç¼«Ê¬¤Ï¸å¼Ô¤òÁªÂò¤·¤Þ¤·¤¿.
¤Þ¤º, ÍøÍѲÄǽ¤Ê¥É¥é¥¤¥Ð¤Î¥Ð¡¼¥¸¥ç¥ó¤ò¥Á¥§¥Ã¥¯¤·¤Þ¤¹.
NVIDIA ¤Î GPU ¤òÀѤó¤Ç¤¤¤ë¤È¥É¥é¥¤¥Ð¤Î¥Ð¡¼¥¸¥ç¥ó¤È CUDA ¤Î¥Ð¡¼¥¸¥ç¥ó¤¬Â·¤ï¤ºÆñµ·¤¹¤ë¤³¤È¤¬Â¿¤¤.
ÆÃ¤Ë Docker ¥¤¥á¡¼¥¸¤Ë²¿¤ò»È¤¨¤Ð...¤ß¤¿¤¤¤ÊÏäˤʤë.
º£²ó¤Ï¤½¤ó¤ÊÏÃ.
¥Ï¥Þ¤Ã¤¿¤Î¤Ï TensorFlow Docker Image ¤ò»È¤ª¤¦¤È¤·¤¿»þ¤Ç¤·¤¿.
¤³¤³¤Ë¤¢¤ë Docker Image ¤ò GPU ¥¿¥°ÉÕ¤¤ÇÆ°¤«¤·¤¿¤È¤³¤í¤É¤¦¤ä¤é GPU ¤¬¸«¤¨¤Æ¤¤¤Ê¤¤¤È¤¤¤¦¤³¤È¤Ëµ¤¤¬¤Ä¤¤¤Æ¤·¤Þ¤¤¤Þ¤·¤¿.
¤È¤¤¤¦¤ï¤±¤Ç tensorflow/tensorflow ¤ò³Îǧ¤¹¤ë¤È...
ARG UBUNTU_VERSION=18.04 |
ARG ARCH= |
ARG CUDA=10.0 |
FROM nvidia/cuda${ARCH:+-$ARCH}:${CUDA}-base-ubuntu${UBUNTU_VERSION} as base |
# ARCH and CUDA are specified again because the FROM directive resets ARGs |
# (but their default value is retained if set previously) |
ARG ARCH |
ARG CUDA |
ARG CUDNN=7.6.2.24-1 |
¤È¤¤¤¦µ½Ò¤Ë.
¤Ä¤Þ¤ê CUDA ¤Î¥Ð¡¼¥¸¥ç¥ó¤Ï 10.0 ¤Ç¥¤¥á¡¼¥¸¤¬¥Ó¥ë¥É¤µ¤ì¤Æ¤¤¤¿¤è¤¦¤Ç¤¹.
¤³¤Î»þ, ¥Þ¥·¥ó¤Î¥Û¥¹¥È¤Î NVRM version ¤Ï 395 ¤Ç, »È¤¨¤ë CUDA version ¤Ï 9.2.
¤Ê¤Î¤Ç Dockerfile ¤Î CUDA ¤ò 9.2 ¤Ç¥Ó¥ë¥É¤¹¤ë¤«, GPU ¤Î¥É¥é¥¤¥Ð¤ò 10.0 ¤¬»È¤¨¤ë¤â¤Î¤Ë¥Ð¡¼¥¸¥ç¥ó¥¢¥Ã¥×¤¹¤ë¤«¤ÎÆóÂò¤Ç¤·¤¿.
Dockerfile ¤Î¥ê¥Ó¥ë¥É¤ÏÌÌÅݤÀ¤Ã¤¿¤Î¤Ç¼«Ê¬¤Ï¸å¼Ô¤òÁªÂò¤·¤Þ¤·¤¿.
¤Þ¤º, ÍøÍѲÄǽ¤Ê¥É¥é¥¤¥Ð¤Î¥Ð¡¼¥¸¥ç¥ó¤ò¥Á¥§¥Ã¥¯¤·¤Þ¤¹.
$ ubuntu-drivers devices== /sys/devices/pci0000:16/0000:16:00.0/0000:17:00.0 ==modalias : pci:v000010DEd00001B06sv00001043sd000085E5bc03sc00i00vendor : NVIDIA Corporationmodel : GP102 [GeForce GTX 1080 Ti]driver : nvidia-driver-415 - third-party freedriver : nvidia-driver-410 - third-party freedriver : nvidia-driver-430 - third-party free recommendeddriver : nvidia-driver-390 - third-party freedriver : xserver-xorg-video-nouveau - distro free builtin
º£²ó¤Ï CUDA versin ¤¬ 10.0 ¤Ë¤·¤¿¤«¤Ã¤¿¤Î¤Ç nvidia-driver-415 ¤òÁªÂò¤·, apt install ¤·¤Þ¤·¤¿.
¤½¤·¤Æ¤Á¤ã¤ó¤È¥¤¥ó¥¹¥È¡¼¥ë¤Ç¤¤Æ¤¤¤ë¤«¥Á¥§¥Ã¥¯.
¤½¤·¤Æ¤Á¤ã¤ó¤È¥¤¥ó¥¹¥È¡¼¥ë¤Ç¤¤Æ¤¤¤ë¤«¥Á¥§¥Ã¥¯.
$ nvidia-container-cli infoNVRM version: 415.27CUDA version: 10.0Device Index: 0Device Minor: 0Model: GeForce GTX 1080 TiBrand: GeForceGPU UUID: GPU-ae5034ee-e52e-37ad-d2b4-19d473f0aeb0Bus Location: 00000000:17:00.0Architecture: 6.1Device Index: 1Device Minor: 1Model: GeForce GTX 1080 TiBrand: GeForceGPU UUID: GPU-210053a3-dbbf-dba5-dca5-5d7e15405c41Bus Location: 00000000:65:00.0Architecture: 6.1
¤³¤Î¥Ö¥í¥°¤Ë¥³¥á¥ó¥È¤¹¤ë¤Ë¤Ï¥í¥°¥¤¥ó¤¬É¬ÍפǤ¹¡£
¤µ¤ó¥í¥°¥¢¥¦¥È
¤³¤Îµ»ö¤Ë¤Ïµö²Ä¥æ¡¼¥¶¤·¤«¥³¥á¥ó¥È¤¬¤Ç¤¤Þ¤»¤ó¡£