2017-02-21 155 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. –