calea inconsecventă a setului de instrumente CUDA /usr vs /usr/lib, construirea tensorflow din sursă

Când folosesc versiunea implicită de tensorflow, primesc Miez de instrucțiuni ilegale aruncat eroare.

Pe baza cercetărilor mele https://stackoverflow.com/questions/60858317/how-to-fix-illegal-instruction-core-dumped & https://github.com/tensorflow/tensorflow/issues/17411 Trebuie să construiesc tensorflow de la sursă.

Am început să construiesc tensorflow din sursă pe baza: https://www.tensorflow.org/install/source

Când aș dori să rulez ./configure, primeam o eroare:

Nu s-a putut găsi nicio versiune cuda.h care să se potrivească cu „10” în niciun subdirector:
    ''
    'include'
    'include/cuda'
    „include/*-linux-gnu”
    „extra/CUPTI/include”
    'include/cuda/CUPTI'
    'local/cuda/extras/CUPTI/include'
de:
    '/lib'
    „/lib/i386-linux-gnu”
    „/lib/x86_64-linux-gnu”
    '/usr'
    „/usr/lib/x86_64-linux-gnu/libfakeroot”

Prin urmare, am instalat cuda toolkit și cudnn pe baza acestei postări: https://towardsdatascience.com/installing-tensorflow-gpu-in-ubuntu-20-04-4ee3ca4cb75d

acum am:

nvcc: driver de compilator NVIDIA (R) Cuda
Drepturi de autor (c) 2005-2019 NVIDIA Corporation
Construit pe Sun_Jul_28_19:07:16_PDT_2019
Instrumente de compilare Cuda, versiunea 10.1, V10.1.243

și Cudddn:

cudnn-10.1-linux-x64-v7.6.5.32 

cuda mea este in:

unde este cuda
cuda: /usr/lib/cuda /usr/include/cuda.h

și nvidia-smi returnează:

Acum m-aș aștepta să pot rula ./configure, primesc următoarele mesaje:

AVERTISMENT: instalarea actuală a bazel nu este o versiune de lansare.
Asigurați-vă că rulați cel puțin bazel 3.7.2
Vă rugăm să specificați locația python. [Implicit este /usr/bin/python3]: 


 S-au găsit căi posibile de bibliotecă Python:
   /usr/lib/python3/dist-packages
    /usr/local/lib/python3.8/dist-packages
   Vă rugăm să introduceți calea dorită a bibliotecii Python de utilizat. Implicit este [/usr/lib/python3/dist-packages]

  Doriți să construiți TensorFlow cu suport ROCm? [da/nu]: 
  Niciun suport ROCm nu va fi activat pentru TensorFlow.

  Doriți să construiți TensorFlow cu suport CUDA? [da/N]: da
  Suportul CUDA va fi activat pentru TensorFlow.

  Doriți să construiți TensorFlow cu suport pentru TensorRT? [da/nu]: 
  Nu va fi activat niciun suport pentru TensorRT pentru TensorFlow.

  Calea setului de instrumente CUDA inconsecventă: /usr vs /usr/lib
  Se solicită configurația CUDA detaliată...


 Vă rugăm să specificați versiunea CUDA SDK pe care doriți să o utilizați. [Lăsați necompletat pentru a implicit CUDA 10]: 


 Vă rugăm să specificați versiunea cuDNN pe care doriți să o utilizați. [Lăsați gol pentru implicit cuDNN 7]: 


 Vă rugăm să specificați versiunea NCCL instalată local pe care doriți să o utilizați. [Lăsați gol pentru utilizare 
 http://github.com/nvidia/nccl]: 


 Vă rugăm să specificați lista de căi de bază, separate prin virgulă, pentru a căuta biblioteci și anteturi CUDA. 
 [Lăsați gol pentru a utiliza valoarea implicită]: 

dar nu pot trece această eroare:

Calea setului de instrumente CUDA inconsecventă: /usr vs /usr/lib Se solicită configurația CUDA detaliată...

Ce s-a întâmplat? cum rezolv aceasta problema?

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
$ sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
$ sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
$ sudo apt-get update
$ sudo apt-get -y install cuda

Sursă: https://developer.nvidia.com/cuda-11.3.0-download-archive.

În cazul meu, aceste instrucțiuni au instalat CUDA sub /usr/local/ și nu direct sub /usr/. În acest fel, procesul de configurare a putut găsi instalarea CUDA.

Sper că asta ajută pe cineva.