tensorflowのconfigureの意味まとめ
最近、tensorflowを高速化するためにコンパイルをしてインストールしようと思った。
その時に、configureファイルを実行するはずだが、そこに書いてある専門用語がちょっと何言ってるかわからない状態だったので、説明。 ちなみに、これは自分がconfigureを実行しようとした時に出てきた物をまとめたものなので他にもあるかもしれない。その時はコメントで教えて。
実際の表示文 | 専門用語 | 意味 |
---|---|---|
Please specify the location of python. | なし | そのまま。pythonのディレクトリを設定する。デフォルトで、tensorflowが見つけたpythonになる。 |
Please input the desired Python library path to use. | なし | そのまま。pythonのディレクトリを変更したときのsite-packages ディレクトリを設定する。普通は自動で検知されるので、変更しない。 |
Do you wish to build TensorFlow with ROCm support? | ROCm | ROCmはnvidiaのCUDAのAMD版的なやつ。普通はいらないのでN 。 |
Do you wish to build TensorFlow with CUDA support? | CUDA | nvidiaGPUを使用するために必要。自分は一応Y 。 |
Do you wish to build TensorFlow with TensorRT support? | TensorRT | nvidiaのやつみたい。よくわからないので自分はN 。わかる方は、コメントで教えてくれると嬉しかったりします。 developer.nvidia.com |
Please specify the CUDA SDK version you want to use. [Leave empty to default to CUDA 10] | なし | CUDAのところをY にした時に表示される。自分はバージョンが10なのでそのままEnter。 |
Please specify the cuDNN version you want to use. [Leave empty to default to cuDNN 7] | なし | cuDNNも7なのでそのままEnter。 |
Please specify the locally installed NCCL version you want to use. [Leave empty to use http://github.com/nvidia/nccl] | NCCL | なんかよくわかんないけど早くなるらしい。そのままEnter。 developer.nvidia.com github.com |
Please specify the comma-separated list of base paths to look for CUDA libraries and headers. | なし | そのまま。普通は空白でOK。 もし、このあとに、CUDAのバージョン指定に戻されるなら、上の方に、足りないファイル名が書いてあるのでそれを探して、もう一度、この質問の時に指定する。 例: Could not find any cublas_api.h matching version '' in any subdirectory |
Please specify a list of comma-separated CUDA compute capabilities you want to build with. You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus. Each capability can be specified as "x.y" or "compute_xy" to include both virtual and binary GPU code, or as "sm_xy" to only include the binary code. Please note that each additional compute capability significantly increases your build time and binary size, and that TensorFlow only supports compute capabilities >= 3.5 [Default is: 6.1] |
いろいろ | 明示されているhttps://developer.nvidia.com/cuda-gpusから、計算能力の数字を調べて入力するっぽい。自分は、1050Tiだったが、乗っていなかったので1050の数字を入力した。 |
Do you want to use clang as CUDA compiler? | なし | clangは使わないのでN 。 |
Please specify which gcc should be used by nvcc as the host compiler. | なし | コンパイラのパスを指定。そのままで大丈夫。 |
Please specify optimization flags to use during compilation when bazel option "--config=opt" is specified | なし | これはそのままでいいと思う。 |
Would you like to interactively configure ./WORKSPACE for Android builds? | なし | androidは使用しないのでN 。 |
これで終わり。あとはビルドして完成。