Google Colab을 사용하여 Vulkan에서 computekernel 실행

배경.

볼칸 머신으로 배우고 싶다

Colab을 통한 개발 및 동작 검증

컨디션

GPU 유형을 미리 선택합니다.
GPU는 트위스트 같고 Tesla K80, Tesla P100, Tesla T4인 것 같아요.
K80이면 안 움직일 수도 있어.
이번에 운이 좋아서 P100에 당첨되었다.P100은 16GB의 메모리가 있어서 정말 좋다.

Vulkan 설정

!apt install vulkan-utils libvulkan-dev

됐어.VK_ICD_FILENAMES 환경 변수 등을 설정할 필요가 없다.
해볼게요.

WARNING: [Loader Message] Code 0 : loader_icd_scan: Can not find 'ICD' object in ICD JSON file /usr/share/vulkan/icd.d/nvidia_layers.json.  Skipping ICD JSON
'DISPLAY' environment variable not set... skipping surface info
error: XDG_RUNTIME_DIR not set in the environment.
==========
VULKANINFO
==========

Vulkan Instance Version: 1.1.126

...

GPU0:
VkPhysicalDeviceProperties:
---------------------------
    apiVersion     = 4198495 (1.1.95)
    driverVersion  = 1754316800 (0x6890c000)
    vendorID       = 0x10de
    deviceID       = 0x15f8
    deviceType     = PHYSICAL_DEVICE_TYPE_DISCRETE_GPU
    deviceName     = Tesla P100-PCIE-16GB

...

    VK_KHR_shader_atomic_int64           : extension revision 1
    VK_KHR_shader_draw_parameters        : extension revision 1
    VK_KHR_shader_float16_int8           : extension revision 1
    VK_KHR_shader_float_controls         : extension revision 1

...

Super cool!
기계 학습에 유익한VK_KHR_shader_float16_int8에도 쓸 수 있다.

이동

vulkan_minimal_컴파일러 컴퓨터가 작동을 시도하고 있습니다.

기타

실시간으로 그리기는 어렵지만 Vulkan에서 도형 그리기에 관해서는 offscreen rendering을 하면 할 수 있을 것 같습니다.
참고: 3D CG 렌더링에 Google Colaboratory가 너무 능숙합니다.
https://qiita.com/stakemura/items/71aeefaa927ab277fcf1

TODO

Colab의 경우 셸 조작이 까다롭기 때문에gitpod self-hosted로 옆에 있는 PC에서 볼칸을 이동시켜 머신러닝 개발https://www.gitpod.io/