เส้นทางชุดเครื่องมือ CUDA ที่ไม่สอดคล้องกัน /usr vs /usr/lib, tensorflow สร้างจากแหล่งที่มา

เมื่อฉันใช้ tensorflow build เริ่มต้น ฉันจะได้รับ ทิ้งแกนคำสั่งที่ผิดกฎหมาย ข้อผิดพลาด.

จากการวิจัยของฉัน https://stackoverflow.com/questions/60858317/how-to-fix-illegal-instruction-core-dumped & https://github.com/tensorflow/tensorflow/issues/17411 ฉันต้องสร้างเทนเซอร์โฟลว์จากแหล่งที่มา

ฉันเริ่มสร้าง tensorflow จากแหล่งที่มาตาม: https://www.tensorflow.org/install/source

เมื่อฉันต้องการเรียกใช้ ./configure ฉันได้รับข้อผิดพลาด:

ไม่พบ cuda.h เวอร์ชันที่ตรงกัน '10' ในไดเร็กทอรีย่อยใดๆ:
    ''
    'รวม'
    'รวม/cuda'
    'รวม/*-linux-gnu'
    'พิเศษ/CUPTI/รวม'
    'รวม/cuda/CUPTI'
    'ท้องถิ่น/cuda/พิเศษ/CUPTI/รวม'
ของ:
    '/ลิบ'
    '/lib/i386-linux-gnu'
    '/lib/x86_64-linux-gnu'
    '/ยูเอส'
    '/usr/lib/x86_64-linux-gnu/libfakeroot'

ดังนั้นฉันจึงติดตั้งชุดเครื่องมือ cuda และ cudnn ตามโพสต์นี้: https://towardsdatascience.com/installing-tensorflow-gpu-in-ubuntu-20-04-4ee3ca4cb75d

ตอนนี้ฉันมี:

nvcc: ไดรเวอร์คอมไพเลอร์ NVIDIA (R) Cuda
ลิขสิทธิ์ (c) 2005-2019 NVIDIA Corporation
สร้างเมื่อ Sun_Jul_28_19:07:16_PDT_2019
เครื่องมือรวบรวม Cuda, รีลีส 10.1, V10.1.243

และ Cudddn:

cudnn-10.1-linux-x64-v7.6.5.32 

cuda ของฉันอยู่ใน:

คูด้าอยู่ไหน
cuda: /usr/lib/cuda /usr/include/cuda.h

และ nvidia-smi ส่งคืน:

ตอนนี้ฉันคาดว่าฉันจะสามารถเรียกใช้ ./configure ฉันได้รับข้อความต่อไปนี้:

คำเตือน: การติดตั้ง bazel ปัจจุบันไม่ใช่เวอร์ชันรีลีส
ตรวจสอบให้แน่ใจว่าคุณใช้ bazel 3.7.2 เป็นอย่างน้อย
โปรดระบุตำแหน่งของงูหลาม [ค่าเริ่มต้นคือ /usr/bin/python3]: 


 พบเส้นทางไลบรารี Python ที่เป็นไปได้:
   /usr/lib/python3/dist-packages
    /usr/local/lib/python3.8/dist-packages
   โปรดป้อนเส้นทางไลบรารี Python ที่ต้องการใช้ ค่าเริ่มต้นคือ [/usr/lib/python3/dist-packages]

  คุณต้องการสร้าง TensorFlow ด้วยการสนับสนุน ROCm หรือไม่ [ใช่/ไม่ใช่]: 
  จะไม่มีการรองรับ ROCm สำหรับ TensorFlow

  คุณต้องการสร้าง TensorFlow ด้วยการสนับสนุน CUDA หรือไม่ [y/N]: ย
  การสนับสนุน CUDA จะเปิดใช้งานสำหรับ TensorFlow

  คุณต้องการสร้าง TensorFlow ด้วยการสนับสนุน TensorRT หรือไม่ [ใช่/ไม่ใช่]: 
  จะไม่มีการรองรับ TensorRT สำหรับ TensorFlow

  เส้นทางชุดเครื่องมือ CUDA ที่ไม่สอดคล้องกัน: /usr vs /usr/lib
  ขอรายละเอียดการกำหนดค่า CUDA...


 โปรดระบุเวอร์ชัน CUDA SDK ที่คุณต้องการใช้ [เว้นว่างไว้เป็นค่าเริ่มต้นเป็น CUDA 10]: 


 โปรดระบุเวอร์ชัน cuDNN ที่คุณต้องการใช้ [เว้นว่างไว้เป็นค่าเริ่มต้นเป็น cuDNN 7]: 


 โปรดระบุเวอร์ชันของ NCCL ที่ติดตั้งในเครื่องที่คุณต้องการใช้ [เว้นว่างไว้ใช้ 
 http://github.com/nvidia/nccl]: 


 โปรดระบุรายการเส้นทางฐานที่คั่นด้วยเครื่องหมายจุลภาคเพื่อค้นหาไลบรารีและส่วนหัวของ CUDA 
 [เว้นว่างไว้เพื่อใช้ค่าเริ่มต้น]: 

แต่ไม่สามารถผ่านข้อผิดพลาดนี้:

เส้นทางชุดเครื่องมือ CUDA ที่ไม่สอดคล้องกัน: /usr vs /usr/lib ขอรายละเอียดการกำหนดค่า CUDA...

เกิดอะไรขึ้น? ฉันจะแก้ไขปัญหานี้ได้อย่างไร

$ 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 อัปเดต
$ sudo apt-get -y ติดตั้ง cuda

แหล่งที่มา: https://developer.nvidia.com/cuda-11.3.0-download-archive.

ในกรณีของฉัน คำแนะนำเหล่านี้ติดตั้ง CUDA ภายใต้ /usr/local/ และไม่อยู่ภายใต้ /usr/ โดยตรง วิธีนี้ทำให้กระบวนการกำหนดค่าสามารถค้นหาการติดตั้ง CUDA ได้

ฉันหวังว่านี่จะช่วยใครซักคน