So, you can install CUDA 10.1 in Ubuntu 20.04 by running, $ sudo apt install nvidia-cuda-toolkitĪfter installing CUDA 10.1, run nvcc -V. The CUDA Toolkit Archive.įor the sake of being verbose, do not try to use 18.10 or 18.04 CUDA 10.1 for Ubuntu 20.04. We are installing CUDA 10.1 because it is the compatible version with TensorFlow GPU. We use the proprietary version over the open source one since CUDA can only operate with the proprietary driver. for CUDA 10.1, the required driver version is ≥ 418.39.
Install TensorFlow 2.0 with GPU support.įirst, ensure that you are using the NVIDIA proprietary driver by going to “Additional Drivers”, and then choosing the appropriate driver, i.e.Install the cuDNN version compatible with CUDA 10.1.Install CUDA 10.1 by installing nvidia-cuda-toolkit.The installation of TensorFlow GPU in Ubuntu 20.04 can be summarized in the following points, So, this post is actually based on the answers given by meetnick and singrium in this related question posted in Ask Ubuntu. Fortunately, I saw some helpful answers online, and now I’m expanding on their answers by adding a bit more explanations. As I expected, it took me a while to work things out in my new system.