인프런 커뮤니티 질문&답변
사이킷런 질문있습니다.
작성
·
594
0
궁금해서 질문 드립니다 LightGBM은 내용에서 GPU를 지원한다고 들었고 혹시 사이킷런 다른 API들은 GPU는 따로 지원하지 않나요 혹시 지원할 수 있는 방법이 따로 있을까요?
답변 1
1
권 철민
지식공유자
안녕하십니까,
아쉽게도 사이킷런은 GPU를 지원하지 않습니다. GPU를 지원하려면 코드를 GPU 지원을 위한 CUDA 코드를 짜야 하는데, 사이킷런은 공식적으로 이런 방향성으로 패키지를 개선하지 않겠다고 얘기해왔습니다.
감사합니다.





