[카테고리:] Data Science
-
Python XGBoost GPU 사용
설치 XGBoost에서 GPU를 사용할 수 없어 상당히 헤맸다.GPU를 사용하려고 하니 아래와 같은 에러가 발생했다. 찾아보니 기존에는 XGBoost에서 GPU를 지원하지 않아 nvidia에서 제공하는 py-xgboost를 사용해야했으나 최근 버전의 XGBoost는 GPU를 지원한다는 모양. [docs] 다만 이를 사용하기 위해서는 XGBoost의 버전과 CUDA 버전이 어느 정도 이상이어야 한다. 필자의 CUDA 버전은 10.0이어서 부랴부랴 11.6으로 버전을 올려주었고, 현재(22.08.12.) 시점에서 conda-forge의 디폴트…