cuda - anaconda 在哪里安装 cudatoolkit 和 cudnn？

Question

我习惯于从 Nvidia 源安装 Cuda 工具包和 cudnn，但刚刚尝试通过 conda 安装以下内容：

conda install cudatoolkit=10.0
conda install -c anaconda cudnn

我现在正在尝试使用 cuda 支持编译FAISS，这需要我在编译时将路径传递给 cuda 安装。

有谁知道 Anaconda 在哪里安装 cuda 工具包？我还想找到 anaconda 安装 cudnn 的位置，以防将来变得相关。其他人可以为这个问题提供的任何帮助都会非常有帮助！

操作系统：Ubuntu 18.04

首先，我尝试使用以下命令搜索 cuda 工具包：

nvcc --version

但这又回来了：

Cuda compilation tools, release 9.1, V9.1.85

版本 9.1，而不是 10.0！所以我尝试了：

locate cuda | grep /cuda$

这返回了几件事：

(3.6) user$ locate cuda | grep /cuda$
/home/dhlab/anaconda3/envs/3.6/include/boost/fiber/cuda
/home/dhlab/anaconda3/envs/3.6/lib/python3.6/site-packages/cupy/cuda
/home/dhlab/anaconda3/envs/3.6/lib/python3.6/site-packages/numba/cuda
/home/dhlab/anaconda3/envs/pixplot/lib/python3.6/site-packages/numba/cuda
/home/dhlab/anaconda3/envs/pixplot/lib/python3.6/site-packages/tensorflow_core/include/external/local_config_cuda/cuda
/home/dhlab/anaconda3/envs/pixplot/lib/python3.6/site-packages/tensorflow_core/include/external/local_config_cuda/cuda/cuda
/home/dhlab/anaconda3/envs/pixplot/lib/python3.6/site-packages/tensorflow_core/include/tensorflow/stream_executor/cuda
/home/dhlab/anaconda3/envs/stylegan/lib/python3.6/site-packages/tensorflow_core/include/external/local_config_cuda/cuda
/home/dhlab/anaconda3/envs/stylegan/lib/python3.6/site-packages/tensorflow_core/include/external/local_config_cuda/cuda/cuda
/home/dhlab/anaconda3/envs/stylegan/lib/python3.6/site-packages/tensorflow_core/include/tensorflow/stream_executor/cuda
/home/dhlab/anaconda3/lib/python3.7/site-packages/numba/cuda
/home/dhlab/anaconda3/pkgs/boost-cpp-1.70.0-h8e57a91_2/include/boost/fiber/cuda
/home/dhlab/anaconda3/pkgs/cupy-7.3.0-py36h273e724_0/lib/python3.6/site-packages/cupy/cuda
/home/dhlab/anaconda3/pkgs/numba-0.45.1-py37h962f231_0/lib/python3.7/site-packages/numba/cuda
/home/dhlab/anaconda3/pkgs/numba-0.49.0rc1-np1.11py3.6h04863e7_gdc1c08865_0/lib/python3.6/site-packages/numba/cuda
/usr/include/thrust/system/cuda
/usr/lib/cuda

我觉得/usr/lib/cuda看起来很诱人，所以我看了看它的内容，但它是空的！然后我以为local_config_cuda里面tensorflow_core可能有线索，所以我进去看了看，在那里我发现了一个头文件，其中包含：

#define TF_CUDA_VERSION "10.0"
#define TF_CUDA_LIB_VERSION "10.0"
#define TF_CUDNN_VERSION "7"

#define TF_CUDA_TOOLKIT_PATH "/usr/local/cuda"

这听起来像是我用 conda 安装的 cuda 工具包的版本，但是 /usr/local/cuda 不存在！

我显然错过了一些东西。如果对安装了 Conda 的 Cuda 有更多经验的人可以帮助我找到 Cuda 工具包，以便我可以告诉 FAISS 在哪里可以找到它，我将非常感激。

标签： cudacondanvidiacudnn