

Caution: Secure BootĬomplicates installation of the NVIDIA driver and is beyond the scope of these instructions. These instructions may work for other Debian-based distros. This section shows how to install CUDA® 11 (TensorFlow >= 2.4.0) on Ubuntuġ6.04 and 18.04. Append its installation directory to the $LD_LIBRARY_PATHĮnvironmental variable: export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64 Install CUDA with apt devel TensorFlow Docker image as a base. Manually install the software requirements listed above, and consider using a

However, if building TensorFlow from source, The apt instructions below are the easiest way to install the required NVIDIA To improve latency and throughput for inference on some models. TensorFlow supports CUDA® 11.2 (TensorFlow >= 2.5.0) The following NVIDIA® software must be installed on your system: You canĮnable compute capabilities by building TensorFlow from source. The TensorFlow package does not contain PTX for your architecture. Note: The error message "Status: device kernel image is invalid" indicates that Packages do not contain PTX code except for the latest supported CUDA®Īrchitecture therefore, TensorFlow fails to load on older GPUs when.

