在安装tensorflow之后，是否可以将tensorflow指向特定版本的CUDA？

Question

我试图通过云端的cuda机器运行tensorflow，我没有sudo权限。

我在virtualenv中安装了tensorflow，但我无法将其指向特定的cuda版本。这只能在从源代码安装时完成吗？

Answer 1

在我使用的Ubuntu系统上（在与运行TensorFlow的终端相同的终端中执行）：

# 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

仅供参考：Is there any point in using CNMeM when running TensorFlow?

Answer 2

通常CUDA_HOME可以解决此类问题。至于TensorFlow高达0.12，homepage提供详细信息。

版本1.0现已上市，文档已更新（此时未重新阅读），因此如果您切换，可能会有额外的信息。