web-dev-qa-db-ja.com

censor8.0でTensorFlow-gpuをインストールする方法は？

私は公式ウェブサイトの指示に従ってそれをインストールしようとしましたが、テンソルフローをインポートするとImportErrorが発生します：

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

コードcat /usr/local/cuda/version.txtを実行します。これは、私のcudaバージョンが8.0.61であることを示しています。

Tensorflowはcuda 9.0を探しているようです。共有gpuサーバーで作業しているため、ルート権限を持っていないため、cudaをアップグレードできません。

Censor 8.0でテンソルフローを動作させる方法はありますか？または他に利用可能な方法はありますか？

ありがとう!!

tensorflowtensorflow-serving
21
inkzk

私が見つけるまで、私は同様の問題に直面していました

https://www.tensorflow.org/install/install_sources#tested_source_configurations

インストールされているcudaバージョンとcudnnバージョンを確認してから、上記のリンクを使用しているものと互換性のあるtensorflow-gpuのバージョンを確認してください。

私はcuda 8とcudnn v5.1をインストールしたので、上記のリンクをチェックすることにより、tensorflow-gpu 1.2.0は互換性があり、インストール後に

pip install tensorflow-gpu==1.2.0

それは私のために働いた。

12
Azam Rafique

CUDA-8のバージョン1.4.1を次のようにインストールする必要があります。

pip install tensorflow-gpu==1.4.1

最新（バージョン1.5）はCUDA-9用です

39
layog