Sebagian besar pesan ini adalah peringatan, bukan kesalahan. Itu hanya berarti bahwa perpustakaan untuk menggunakan GPU Nvidia tidak diinstal, tetapi Anda tidak perlu memiliki GPU Nvidia untuk menggunakan Tensorflow sehingga Anda tidak memerlukan perpustakaan ini. Komentar oleh jakub memberi tahu cara mematikan peringatan:
export TF_CPP_MIN_LOG_LEVEL="2"
Namun, saya juga menjalankan Tensorflow tanpa hal-hal Nvidia dan ada satu pesan lagi yang merupakan kesalahan, bukan peringatan:
2020-04-10 10:04:13.365696: E tensorflow/stream_executor/cuda/cuda_driver.cc:351] failed call to cuInit: UNKNOWN ERROR (303)
Seharusnya tidak relevan karena itu juga mengacu pada cuda, yaitu untuk Nvidia. Namun sepertinya itu bukan kesalahan fatal.
Saya mendapat peringatan ini sebagai akibat dari pembaruan (tidak disengaja) dari paket libvnifer6. Itu diperbarui menjadi 6.0.1-1+cuda10.2
sedangkan instalasi asli menggunakan 6.0.1-1+cuda10.1
.
Setelah saya menghapus paket yang merujuk cuda10.2
dan dijalankan ulang
sudo apt-get install -y --no-install-recommends libnvinfer6=6.0.1-1+cuda10.1 \
libnvinfer-dev=6.0.1-1+cuda10.1 \
libnvinfer-plugin6=6.0.1-1+cuda10.1
peringatan ini hilang.
Ini adalah peringatan, bukan kesalahan. Anda masih dapat menggunakan TensorFlow. Pustaka bersama libnvinfer
dan libnvinfer_plugin
bersifat opsional dan hanya diperlukan jika Anda menggunakan kemampuan TensorRT nvidia.
Petunjuk penginstalan TensorFlow mencantumkan dependensi GPU:
Perangkat lunak NVIDIA® berikut harus diinstal di sistem Anda:
- Driver GPU NVIDIA® —CUDA 10.1 memerlukan 418.x atau lebih tinggi.
- CUDA® Toolkit —TensorFlow mendukung CUDA 10.1 (TensorFlow>=2.1.0)
- CUPTI dikirimkan bersama CUDA Toolkit.
- cuDNN SDK (>=7.6)
- (Opsional) TensorRT 6.0 untuk meningkatkan latensi dan throughput untuk inferensi pada beberapa model.
Anda dapat menginstalnya di Ubuntu 18.04 dengan perintah berikut (diambil dari dokumentasi TensorFlow):
# Add NVIDIA package repositories
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.1.243-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804_10.1.243-1_amd64.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo apt-get update
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo apt install ./nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo apt-get update
# Install NVIDIA driver
sudo apt-get install --no-install-recommends nvidia-driver-430
# Reboot. Check that GPUs are visible using the command: nvidia-smi
# Install development and runtime libraries (~4GB)
sudo apt-get install --no-install-recommends \
cuda-10-1 \
libcudnn7=7.6.4.38-1+cuda10.1 \
libcudnn7-dev=7.6.4.38-1+cuda10.1
# Install TensorRT. Requires that libcudnn7 is installed above.
sudo apt-get install -y --no-install-recommends libnvinfer6=6.0.1-1+cuda10.1 \
libnvinfer-dev=6.0.1-1+cuda10.1 \
libnvinfer-plugin6=6.0.1-1+cuda10.1