2017-02-21 154 views

回答

2

在我用我的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?

+1

设置LD_LIBRARY_PATH工作。谢谢! –

0

通常CUDA_HOME允许解决这样的问题。至于0.12的TensorFlow,homepage给出了详细信息。

版本1.0现在可用,并且文档已更新(此时不重新阅读),因此如果切换,可能会有额外的信息。

+0

我试着设置CUDA_HOME变量,但没有奏效。我安装了多个cuda版本。我需要将它指向版本8. –