Skip to content

Ubuntu 18.04 with GPU

Peter Leonard edited this page Dec 18, 2019 · 11 revisions
  1. Start with a blank hard drive and this set of recommendations from Puget Systems.

  2. In step 6 of the Puget Systems walkthrough, install nvidia-driver-435 (current as of December 2019) instead of nvidia-390.

  3. Once your graphical desktop is set up and configured, check the Additional Drivers tab of Software & Updates to ensure that you are using Nvidia-435 (it should be described by Ubuntu as "proprietary".)

  4. Install anaconda.

  5. Create a new environment for pixplot:

conda create --name pixppot python=3.6 (Tensorflow will not yet work easily with python 3.7)

conda activate pixplot

  1. Get the basics of CUDA and cuDNN set up:

conda install cudatoolkit=10.0

conda install -c anaconda cudnn=7.6.4

  1. Install pixplot and supporting packages:

pip install pixplot (will install the CPU version of TensorFlow)

pip install tensorflow-gpu==1.15.0 (the GPU-accelerated version)

pip install image_datasets (a helper utility to easily grab pixplot-ready datasets)

  1. Get a sample dataset:

python

import image_datasets

image_datasets.oslomini.download()

(now exit python)

  1. Test the installation:

time pixplot --images="datasets/oslomini/images/*.jpg" --metadata="datasets/oslomini/metadata.csv"

In another terminal window, you can run watch -n 0.1 nvidia-smi to monitor how TensorFlow uses your GPU.