テンソルフローをインストールした後にテンソルフローを特定のバージョンのCUDAにポイントすることは可能ですか？

Question

私はsudoの権限を持たないクラウド上で、クンダマシンでテンソルフローを実行しようとしています。

私はテンソルフローをvirtualenvにインストールしましたが、特定のcudaバージョンに向けて指すことはできません。ソースからインストールする場合にのみこれを実行できますか？

Answer 1

：

# CUDA 7.5: 
export PATH=/usr/local/cuda-7.5/bin:$PATH 
export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH 

# CUDA 8.0: 
export PATH=/usr/local/cuda-8.0/bin:$PATH 
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH 

あなたはまたCuDNNとCNMeMを指すようにしたいことがあります。

# CuDNN: 
export LD_LIBRARY_PATH=/scratch1/francky/cudnn/cuda/lib64:$LD_LIBRARY_PATH 
export CPATH=/scratch1/francky/cudnn/cuda/include:$CPATH 
export LIBRARY_PATH=/scratch1/francky/cudnn/cuda/lib64:$LD_LIBRARY_PATH 

# CNMeM: 
export LD_LIBRARY_PATH=/scratch1/francky/cudnn/cnmem/build:$LD_LIBRARY_PATH 
export CPATH=/scratch1/francky/cudnn/cnmem/include:$CPATH 
export LIBRARY_PATH=/scratch1/francky/cudnn/cnmem/build:$LD_LIBRARY_PATH 

FYI：Is there any point in using CNMeM when running TensorFlow?

Answer 2

通常、CUDA_HOMEは、この種の問題を解決します。 TensorFlowが0.12までは、homepageが詳細情報を提供します。

バージョン1.0は現在利用可能で、ドキュメントは更新されています（この時点では再読み込みされません）。私は（あなたがTensorFlowを実行する一つとして同じ端末で実行する）を使用し、私のUbuntuシステムで