Question

过去，安装tensorflow-gpu需单独安装CUDA和cuDNN，并将LD_LIBRARY_PATH和CUDA_HOME的路径添加到环境中。

现在，一个简单的conda install tensorflow-gpu==1.9就可以处理所有事情。从环境中删除CUDA_HOME和LD_LIBRARY_PATH对tensorflow-gpu毫无影响。

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

Answer 1

在安装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)

您唯一需要的是libcuda.so.1，一旦安装了cuda驱动程序，通常即可在标准库搜索目录列表中找到它。

Answer 2

就我而言，以下命令会自动处理它：

$ sudo apt install nvidia-cuda-toolkit

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

2 个答案: