从源建立张量流时Cuda位置错误

Question

我正在从源代码构建tensorlfow，但是当我运行configure时，我遇到了cuda路径时的错误。这是它要求的：

请指定安装CUDA 9.1工具包的位置。有关更多详细信息，请参阅README.md。 [默认为/ usr / local / cuda]：

我的cuda安装来自ubuntu软件包。

我认为cuda已经安装好了。我跑的时候

nvcc --version

我输出为

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85

想知道如何找出cuda toolkit的位置。我尝试使用/ usr / lib / cuda，/ usr / local / cuda，/ usr / lib / nvidia-cuda-toolkit，/ usr / lib / nvidia-cuda-toolkit / bin。这些都不起作用。

我收到错误，因为“NameError：全局名称'cuda_toolkit_path_full'未定义”

找到安装CUDA 9.1工具包的正确位置位置的任何指针都会有所帮助。谢谢。

Answer 1

我给了/ usr目录，之后它停止显示该错误。