我想通過雲中的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工作。謝謝! –