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

我试图在云上的cuda机器上运行tensorflow，而我没有sudo特权

---

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

通常

CUDA_HOME

允许解决此类问题。对于高达0.12的TensorFlow，提供了详细信息

---

1.0版现在可用，文档已经更新（此时不再重读），因此如果您切换，可能会有额外的信息。

在我使用的Ubuntu系统上（将在运行TensorFlow的同一个终端上执行）：

您还可以指向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

---

仅供参考：

我尝试设置CUDA_HOME变量，但无效。我安装了多个cuda版本。我需要将它指向版本8。是否有任何方法可以将其设置为在启动virtualenv时自动运行？就像把它放进去一样。bashrc可以。。。

# 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