![]() ![]() ![]() Prior to starting CUDA download and installation, make the checks suggested here on the Nvidia CUDA website: Step 2: Pre-CUDA installation: check existing installations NVIDIA GPU driver 390.132 (CUDA 9.0 requires 384.x or greater).cuDNN version of 7.2, required for Tensorflow version 1.12.I therefore already have the following installed on this machine prior to completing the new steps below: When I previously installed Tensorflow on this Ubuntu 18.04 machine, only Tensorflow 1.12/CUDA 9 was available and CUDA 10 was not yet compatible with Tensorflow. (Optional) TensorRT 6.0 to improve latency and throughput for inference on some models.I would like to be able to install various versions of Tensorflow (with GPU support) between 1.13–2.1, so CUDA 10.1 is definitely required Tensorflow 1.13 and above requires CUDA 10. CUDA® Toolkit - TensorFlow supports CUDA 10.1 (TensorFlow >= 2.1.0).NVIDIA GPU drivers - CUDA 10.1 requires 418.x or higher.Therefore before moving through the steps of installing an Nvidia driver, CUDA, cuDNN and then Tensorflow 2.1, I’m “ beginning with the end in mind” and first checking the correct software versions compatible with my target version of Tensorflow.Īccording to the Tensorflow website and CUDA installation guide: The version of Tensorflow you select will determine the compatible versions of CUDA, cuDNN, compiler, toolchain and the Nvidia driver versions to install. Step 1: Checking which versions of drivers and software to install for mutual compatibility with Tensorflow 2.1 In doing so, in my case this involves also handling my current installations of Nvidia drivers, CUDA, cuDNN, and Tensorflow (details of which are set out at Step 1). In this Part 4 of the series, I am installing drivers for the Nvidia GPU which are compatible with the version of CUDA Toolkit, cuDNN and Tensorflow I wish to install on Ubuntu 18.04, namely Tensorflow 2.1 - this requires CUDA 10.1 or above. In Part 3, I wiped Windows 10 from my PC and installed Ubuntu 18.04 LTS from a bootable DVD. Part 2 of the series covered the installation of CUDA, cuDNN and Tensorflow on Windows 10. In Part 1 of this series, I discussed how you can upgrade your PC hardware to incorporate a CUDA Toolkit compatible graphics processing card and I installed an Nvidia GTX 1060 6GB. Pimp Up your PC for Deep Learning Series - Part 4 Introduction ![]()
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |