状況
- Docker(in WSL環境)でLightGBMをGPUを使おうとした際に発生したエラー
"lightgbm.basic.LightGBMError", 'No OpenCL device found'
結論
コンテナ内でGPUが使える状況になっていなかった。
nvidia-smi
コマンドで確認のこと
今回はDockerComposeを使っていたが、オプション指定がうまくなかった模様。下記で通った(countが省略できなかった)
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [gpu]