我想通过云中的cuda机器上运行tensorflow,但我没有sudo权限。在tensorflow安装后,是否可以将tensorflow指向特定版本的CUDA?
我将tensorflow安装在virtualenv中,但我无法将其指向特定的cuda版本。这只能从源代码安装完成吗?
我想通过云中的cuda机器上运行tensorflow,但我没有sudo权限。在tensorflow安装后,是否可以将tensorflow指向特定版本的CUDA?
我将tensorflow安装在virtualenv中,但我无法将其指向特定的cuda版本。这只能从源代码安装完成吗?
在我用我的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?
通常CUDA_HOME
允许解决这样的问题。至于0.12的TensorFlow,homepage给出了详细信息。
版本1.0现在可用,并且文档已更新(此时不重新阅读),因此如果切换,可能会有额外的信息。
我试着设置CUDA_HOME变量,但没有奏效。我安装了多个cuda版本。我需要将它指向版本8. –
设置LD_LIBRARY_PATH工作。谢谢! –