web-dev-qa-db-ja.com

TensorflowのCUDA_HOMEパス

Tensorflow Linuxインストール手順 言う:

NVIDIAのドキュメントの説明に従って、CUDA_HOME環境変数を作成してください。

CuDNN v6のNVIDIAの指示または NVIDIA CUDA Toolkitのインストール手順 にCUDA_HOMEの記述が見つかりません。 Linuxでこの変数を設定する方法を知っている人はいますか?

20
Ron Cohen

ターミナルで次のコマンドを実行します。

export CUDA_HOME=/usr/local/cuda-X.X

X.Xをバージョン番号の最初の2桁で置き換えます(例:nvcc --version)。

16
Dmitry