日々精進

新しく学んだことを書き留めていきます

CenterNetをインストールしようとすると「nvcc: command not found」エラー

CenterNetのインストール手順にmake.shを実行する手順があるが、これを実行すると「nvcc: command not found」エラーが発生した。 https://github.com/xingyizhou/CenterNet/blob/master/readme/INSTALL.md

nvccはcudaのコンパイラで、cuda toolkitに含まれるらしい。 なのでcuda toolkitをインストールしてみたが、やっぱりnvccコマンドが使えない。

調べてるうちに、ビルド環境のコンテナのベースイメージがcuda base imageだからnvccが入ってないということがわかった。 devel imageは開発用のイメージでnvccも含まれるらしい。

なのでベースイメージを nvidia/cuda:9.0-devel-ubuntu16.04 にしたら直った。

参考:

vasteelab.com