带conda的Tensorflow gpu：CUDA_HOME在哪里指定？

以前，安装

tensorflow gpu

需要分别安装CUDA和cuDNN，并将路径添加到

LD_LIBRARY_路径

和

CUDA_HOME

环境中

现在，一个简单的

conda安装tensorflow gpu==1.9

就能解决所有问题。从环境中删除

CUDA_主路径

和

LD_库路径

对tensorflow gpu没有任何影响

---

问题：使用anaconda安装TensorFlow时，为TensorFlow指定的CUDA路径在哪里？

安装TensorFlow gpu时，它会安装另外两个conda软件包：

cudatoolkit:       9.0-h13b8566_0          
cudnn:             7.1.2-cuda9.0_0         

如果仔细观察tensorflow动态共享对象，它会使用RPATH在Linux上获取以下库：

(tflow) $ ldd $CONDA_PREFIX/lib/python3.6/site-packages/tensorflow/python/_pywrap_tensorflow_internal.so | grep -E 'cufft|curand'
    libcufft.so.9.0 => /tmp/wani.1533146686/tflow/lib/python3.6/site-packages/tensorflow/python/../../../../libcufft.so.9.0 (0x00007fbb9454a000)
    libcurand.so.9.0 => /tmp/wani.1533146686/tflow/lib/python3.6/site-packages/tensorflow/python/../../../../libcurand.so.9.0 (0x00007fbb905e4000)

(tflow) $ ldd $CONDA_PREFIX/lib/python3.6/site-packages/tensorflow/python/_pywrap_tensorflow_internal.so | grep cudnn
    libcudnn.so.7 => /tmp/wani.1533146686/tflow/lib/python3.6/site-packages/tensorflow/python/../../../../libcudnn.so.7 (0x00007fd73b55d000)

---

安装cuda驱动程序后，您只需要

libcuda.so.1

，它通常在库的标准搜索目录列表中可用。

在我的情况下，以下命令会自动处理它：

$ sudo apt install nvidia-cuda-toolkit