본문 바로가기
리눅스

WSL 2에서 GPU accelerated machine learning training(GPU 사용하기) - 1

by 머리올리자 2021. 1. 3.

Windows Insider SDK는 WSL 2 인스턴스 내에서 GPU 하드웨어 가속을 위해 NVIDIA CUDA를 사용하는 기존 ML 도구, 라이브러리 및 인기있는 프레임 워크 실행을 지원합니다. 여기에는 PyTorch 및 TensorFlow는 물론 기본 Linux 환경에서 사용 가능한 모든 Docker 및 NVIDIA Container Toolkit 지원이 포함

 

다음 기능은 Windows 10 시험판 버전에서 사용할 수 있으며 변경 될 수 있음

 

이 미리보기를 사용하려면 Windows 참가자 프로그램에 등록  (아래 링크)

insider.windows.com/en-us/getting-started#register

 

Windows Insider Program

Become an Insider: be one of the first to explore new Windows features for you and your business or use the latest Windows SDK to build great apps.

insider.windows.com

등록 완료

 

 

 

그런 다음이 지침에 따라 최신 Insider 빌드를 설치. (아래 링크)

insider.windows.com/en-us/getting-started#install

 

Windows Insider Program

Become an Insider: be one of the first to explore new Windows features for you and your business or use the latest Windows SDK to build great apps.

insider.windows.com

 

 

1. Settings >.... 의 링크를 클릭하면

선택형 진단 데이터 클릭

계정 추가

개발자 채널

지금 다시 시작

 

피드백 및 진단에서 모든 설정을 켬

윈도우 업데이트 들어가면 업데이트 항목이 뜸

 

Run 명령 (Windows 로고 키 + R)을 통해 winver를 실행하여 빌드 버전 번호를 확인

 

developer.nvidia.com/cuda/wsl

 

GPU in Windows Subsystem for Linux (WSL)

CUDA on Windows Subsystem for Linux (WSL) - Public Preview Microsoft Windows is a ubiquitous platform for enterprise, business, and personal computing systems. However, industry AI tools, models, frameworks, and libraries are predominantly available on Lin

developer.nvidia.com

위 링크에서 NVIDIA CUDA-enabled driver for WSL 다운 

WSL 2와 Linux 배포(Ubuntu나 Debian 과 같은)가 설치되어 있어야 한다.

드라이버를 설치 완료하고

 

powershell을 열어 wsl cat /proc/version을 입력했을 때 리눅스 커널 버전이 4.19.121 이상이어야 한다.

 

너무 길어 다음 페이지에 넘겨서 확인해보자

 

참고

docs.microsoft.com/ko-kr/windows/win32/direct3d12/gpu-cuda-in-wsl

 

Enable NVIDIA CUDA in WSL 2 - Win32 apps

Enable the NVIDIA CUDA preview on the Windows Subsystem for Linux

docs.microsoft.com