How to Install TensorFlow GPU on Linux

Profile Picture
- Published on May 16, 2019馃審 Public

This tutorial will show you how to install tensorflow gpu on linux/ubuntu. Installing tensorflow gpu requires that you have a CUDA enabled gpu (typically a GTX1050ti 0r higher). Tensorflow website: Commands: # Add NVIDIA package repositories wget sudo dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb sudo apt-key adv --fetch-keys sudo apt-get update wget 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-410 # 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-0 \ libcudnn7= \ libcudnn7-dev= # Install TensorRT. Requires that libcudnn7 is installed above. sudo apt-get update && \ sudo apt-get install nvinfer-runtime-trt-repo-ubuntu1804-5.0.2-ga-cuda10.0 \ && sudo apt-get update \ && sudo apt-get install -y --no-install-recommends libnvinfer-dev=5.0.2-1+cuda10.0 Want a sneak peak into my life? Follow my Instagram @tech_with_tim where I'm going to be filming a video each morning sharing my goals for the day and what I have planned: ** WEBSITE: One-Time Donations: Support the Channel: Podcast: Twitter: Join my discord server: ** Please leave a LIKE and SUBSCRIBE for more content! Tags: - Tech With Tim - How to Install Tensorflow - Python Tutorials