-
Notifications
You must be signed in to change notification settings - Fork 141
Ubuntu 18.04 with GPU
-
Start with a blank hard drive and this set of recommendations from Puget Systems.
-
In step 6 of the Puget Systems walkthrough, install
nvidia-driver-435
(current as of December 2019) instead ofnvidia-390
. -
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".)
-
Install anaconda.
-
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
- Get the basics of CUDA and cuDNN set up:
conda install cudatoolkit=10.0
conda install -c anaconda cudnn=7.6.4
- 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)
- Get a sample dataset:
python
import image_datasets
image_datasets.oslomini.download()
(now exit python)
- 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.