해당 내용은 CentOS 7 을 기준으로 작성
구조
호스트의 GPU 를 docker 컨테이너에서 사용할 수 있음
CUDA
NVIDIA 에서는 cuda 라이브러리가 설치된 이미지를 배포 하고 있음
https://hub.docker.com/r/nvidia/cuda/
Ubuntu, CentOS 만 지원
OpenCL
OpenCL 라이브러리가 설치된 이미지도 있음
https://hub.docker.com/r/nvidia/opencl/
용량이 훨씬 작고 마찬가지로 Ubuntu, CentOS 만 지원
설치
docker 설치
xxxxxxxxxx$ curl -fsSL https://get.docker.com/ | sh
driver 설치
CUDA Toolkit 다운로드
https://developer.nvidia.com/cuda-downloads
CUDA Toolkit 설치
xxxxxxxxxx$ rpm -ivh cuda-repo-rhel7-9.1.85-1.x86_64.rpm$ yum install cudaCUDA Toolkit 에는 드라이버를 포함하고 있으므로 호스트에 설치되어 있는 드라이버는 삭제하고 진행해야 함
nvidia-docker 설치
https://github.com/NVIDIA/nvidia-docker/wiki
xxxxxxxxxx$ distribution=$(. /etc/os-release;echo $ID$VERSION_ID)$ curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo$ yum install -y nvidia-docker2$ systemctl restart docker
docker 사용
CUDA
x
$ docker run --runtime=nvidia -ti --rm nvidia/cuda:9.1-devel-centos7 /bin/bash$ nvidia-smiWed May 16 10:25:56 2018 +-----------------------------------------------------------------------------+| NVIDIA-SMI 390.30 Driver Version: 390.30 ||-------------------------------+----------------------+----------------------+| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC || Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. ||===============================+======================+======================|| 0 GeForce GT 710 Off | 00000000:01:00.0 N/A | N/A || 40% 29C P0 N/A / N/A | 0MiB / 980MiB | N/A Default |+-------------------------------+----------------------+----------------------+| 1 Tesla K40c Off | 00000000:02:00.0 Off | 0 || 23% 30C P0 69W / 235W | 0MiB / 11441MiB | 75% Default |+-------------------------------+----------------------+----------------------+ +-----------------------------------------------------------------------------+| Processes: GPU Memory || GPU PID Type Process name Usage ||=============================================================================|| 0 Not Supported |+-----------------------------------------------------------------------------
OpenCL
x
$ docker run --runtime=nvidia -ti --rm nvidia/opencl:devel-centos7 /bin/bash$ clinfoNumber of platforms 1 Platform Name NVIDIA CUDA Platform Vendor NVIDIA Corporation Platform Version OpenCL 1.2 CUDA 9.1.84 Platform Profile FULL_PROFILE Platform Extensions cl_khr_global_int32_base_atomics cl_khr_global_int32_extended_atomics cl_khr_local_int32_base_atomics cl_khr_local_int32_extended_atomics cl_khr_fp64 cl_khr_byte_addressable_store cl_khr_icd cl_khr_gl_sharing cl_nv_compiler_options cl_nv_device_attribute_query cl_nv_pragma_unroll cl_nv_copy_opts cl_nv_create_buffer Platform Extensions function suffix NV Platform Name NVIDIA CUDANumber of devices 2 Device Name Tesla K40c Device Vendor NVIDIA Corporation Device Vendor ID 0x10de Device Version OpenCL 1.2 CUDA Driver Version 390.30 Device OpenCL C Version OpenCL C 1.2 Device Type GPU Device Available Yes Device Profile FULL_PROFILE Device Topology (NV) PCI-E, 02:00.0 Max compute units 15 Max clock frequency 745MHz Compute Capability (NV) 3.5 Device Partition (core) Max number of sub-devices 1 Supported partition types None Max work item dimensions 3 Max work item sizes 1024x1024x64 Max work group size 1024 Compiler Available Yes Linker Available Yes Preferred work group size multiple 32 Warp size (NV) 32 Preferred / native vector sizes char 1 / 1 short 1 / 1 int 1 / 1 long 1 / 1 half 0 / 0 (n/a) float 1 / 1 double 1 / 1 (cl_khr_fp64) Half-precision Floating-point support (n/a) Single-precision Floating-point support (core) Denormals Yes Infinity and NANs Yes Round to nearest Yes Round to zero Yes Round to infinity Yes IEEE754-2008 fused multiply-add Yes Support is emulated in software No Correctly-rounded divide and sqrt operations Yes Double-precision Floating-point support (cl_khr_fp64) Denormals Yes Infinity and NANs Yes Round to nearest Yes Round to zero Yes Round to infinity Yes IEEE754-2008 fused multiply-add Yes Support is emulated in software No Correctly-rounded divide and sqrt operations No Address bits 64, Little-Endian Global memory size 11996954624 (11.17GiB) Error Correction support Yes Max memory allocation 2999238656 (2.793GiB) Unified memory for Host and Device No Integrated memory (NV) No Minimum alignment for any data type 128 bytes Alignment of base address 4096 bits (512 bytes) Global Memory cache type Read/Write Global Memory cache size 245760 (240KiB) Global Memory cache line 128 bytes Image support Yes Max number of samplers per kernel 32 Max size for 1D images from buffer 134217728 pixels Max 1D or 2D image array size 2048 images Max 2D image size 16384x16384 pixels Max 3D image size 4096x4096x4096 pixels Max number of read image args 256 Max number of write image args 16 Local memory type Local Local memory size 49152 (48KiB) Registers per block (NV) 65536 Max constant buffer size 65536 (64KiB) Max number of constant args 9 Max size of kernel argument 4352 (4.25KiB) Queue properties Out-of-order execution Yes Profiling Yes Prefer user sync for interop No Profiling timer resolution 1000ns Execution capabilities Run OpenCL kernels Yes Run native kernels No Kernel execution timeout (NV) No Concurrent copy and kernel execution (NV) Yes Number of async copy engines 2 printf() buffer size 1048576 (1024KiB) Built-in kernels Device Extensions cl_khr_global_int32_base_atomics cl_khr_global_int32_extended_atomics cl_khr_local_int32_base_atomics cl_khr_local_int32_extended_atomics cl_khr_fp64 cl_khr_byte_addressable_store cl_khr_icd cl_khr_gl_sharing cl_nv_compiler_options cl_nv_device_attribute_query cl_nv_pragma_unroll cl_nv_copy_opts cl_nv_create_buffer Device Name GeForce GT 710 Device Vendor NVIDIA Corporation Device Vendor ID 0x10de Device Version OpenCL 1.2 CUDA Driver Version 390.30 Device OpenCL C Version OpenCL C 1.2 Device Type GPU Device Available Yes Device Profile FULL_PROFILE Device Topology (NV) PCI-E, 01:00.0 Max compute units 1 Max clock frequency 954MHz Compute Capability (NV) 3.5 Device Partition (core) Max number of sub-devices 1 Supported partition types None Max work item dimensions 3 Max work item sizes 1024x1024x64 Max work group size 1024 Compiler Available Yes Linker Available Yes Preferred work group size multiple 32 Warp size (NV) 32 Preferred / native vector sizes char 1 / 1 short 1 / 1 int 1 / 1 long 1 / 1 half 0 / 0 (n/a) float 1 / 1 double 1 / 1 (cl_khr_fp64) Half-precision Floating-point support (n/a) Single-precision Floating-point support (core) Denormals Yes Infinity and NANs Yes Round to nearest Yes Round to zero Yes Round to infinity Yes IEEE754-2008 fused multiply-add Yes Support is emulated in software No Correctly-rounded divide and sqrt operations Yes Double-precision Floating-point support (cl_khr_fp64) Denormals Yes Infinity and NANs Yes Round to nearest Yes Round to zero Yes Round to infinity Yes IEEE754-2008 fused multiply-add Yes Support is emulated in software No Correctly-rounded divide and sqrt operations No Address bits 64, Little-Endian Global memory size 1028587520 (980.9MiB) Error Correction support No Max memory allocation 257146880 (245.2MiB) Unified memory for Host and Device No Integrated memory (NV) No Minimum alignment for any data type 128 bytes Alignment of base address 4096 bits (512 bytes) Global Memory cache type Read/Write Global Memory cache size 16384 (16KiB) Global Memory cache line 128 bytes Image support Yes Max number of samplers per kernel 32 Max size for 1D images from buffer 134217728 pixels Max 1D or 2D image array size 2048 images Max 2D image size 16384x16384 pixels Max 3D image size 4096x4096x4096 pixels Max number of read image args 256 Max number of write image args 16 Local memory type Local Local memory size 49152 (48KiB) Registers per block (NV) 65536 Max constant buffer size 65536 (64KiB) Max number of constant args 9 Max size of kernel argument 4352 (4.25KiB) Queue properties Out-of-order execution Yes Profiling Yes Prefer user sync for interop No Profiling timer resolution 1000ns Execution capabilities Run OpenCL kernels Yes Run native kernels No Kernel execution timeout (NV) No Concurrent copy and kernel execution (NV) Yes Number of async copy engines 1 printf() buffer size 1048576 (1024KiB) Built-in kernels Device Extensions cl_khr_global_int32_base_atomics cl_khr_global_int32_extended_atomics cl_khr_local_int32_base_atomics cl_khr_local_int32_extended_atomics cl_khr_fp64 cl_khr_byte_addressable_store cl_khr_icd cl_khr_gl_sharing cl_nv_compiler_options cl_nv_device_attribute_query cl_nv_pragma_unroll cl_nv_copy_opts cl_nv_create_buffer NULL platform behavior clGetPlatformInfo(NULL, CL_PLATFORM_NAME, ...) NVIDIA CUDA clGetDeviceIDs(NULL, CL_DEVICE_TYPE_ALL, ...) Success [NV] clCreateContext(NULL, ...) [default] Success [NV] clCreateContextFromType(NULL, CL_DEVICE_TYPE_CPU) No devices found in platform clCreateContextFromType(NULL, CL_DEVICE_TYPE_GPU) No platform clCreateContextFromType(NULL, CL_DEVICE_TYPE_ACCELERATOR) No devices found in platform clCreateContextFromType(NULL, CL_DEVICE_TYPE_CUSTOM) No devices found in platform clCreateContextFromType(NULL, CL_DEVICE_TYPE_ALL) No platform ICD loader properties ICD loader Name OpenCL ICD Loader ICD loader Vendor OCL Icd free software ICD loader Version 2.2.12 ICD loader Profile OpenCL 2.2 NOTE: your OpenCL library declares to support OpenCL 2.2, but it seems to support up to OpenCL 2.1 only.
'가상화 > docker' 카테고리의 다른 글
| docker registry 생성 자동화 (0) | 2018.09.11 |
|---|---|
| docker 사용 (0) | 2018.09.11 |
| docker 한글 지원 (1) | 2018.09.11 |
| docker alpine linux (0) | 2018.05.28 |
| docker 설치 후 service 시작 실패 (0) | 2018.05.28 |