Ошибка определения местоположения Cuda при построении tensorflow из исходного кода

#python #ubuntu #tensorflow

#python #ubuntu #tensorflow

Вопрос:

Я создаю tensorlfow из исходного кода, но я сталкиваюсь с ошибкой, когда речь идет о пути cuda во время запуска configure. Это то, что он запрашивает:

Пожалуйста, укажите местоположение, в котором установлен CUDA 9.1 toolkit. Обратитесь к README.md для получения более подробной информации. [По умолчанию используется /usr/local/cuda]:

Моя установка cuda была из пакетов ubutnu.

Я думаю, что 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. Ни один из них не работает.

Я получаю ошибку типа «Ошибка имени: глобальное имя ‘cuda_toolkit_path_full’ не определено «

Любые указания для определения правильного местоположения, в котором установлен CUDA 9.1 toolkit, будут полезны. Спасибо.

Комментарии:

1. Какую версию TensorFlow вы пытаетесь скомпилировать из исходного кода?

2. Я использую версию 1.11

Ответ №1:

Я указал каталог / usr, после чего он перестал показывать эту ошибку.