openposeを実行したときのエラーについての忘備録
openposeについては下記参照.
https://github.com/CMU-Perceptual-Computing-Lab/openpose
###発生したエラー
For the "CUBLAS_STATUS_ALLOC_FAILED" problem, it's because the demo wants to use all visible devices and thus caused crash.
GPU関連の設定が怪しいがpytorchなどの別のプログラムでは普通にGPUを使えていたことは確認済み.
openposeの実行時にCUDA_VISIBLE_DEVICES=2
を加えることで無事に実行できた.
CUDAに使用するデバイスの設定ミスが原因...?
CUDA_VISIBLE_DEVICES=2 ./build/examples/openpose/openpose.bin --net_resolution "160x80" --video examples/media/video.avi
毎回追加する場合が面倒な場合は~/.bashrc
内に下記の1行を追加する
~/.bashrc
export CUDA_VISIBLE_DEVICES=2
~/.bashrc
を変更した後はターミナルを起動し直す,または下記のコマンドを実行
source ~/.bashrc
###おまけ
openposeをrosで利用
https://github.com/firephinx/openpose_ros