Ubuntu基于Docker搭建TensorFlow+Pycharm环境

Setup

Prerequisites

• Ubuntu (preferably 16.04 or up) . CentOS is also supported but may need some extra settings.
• Nvidia GPU (optional but strongly recommended)
• Docker Hub account
• Make sure your account has root access. Refer to here for instructions on how to grant sudo access.

Setup Docker Environment on Ubuntu host

Run the following script as a file with bash under sudo.(You can skip this step if you've had it setup already).

• Install Nvidia Driver

You can install nvidia driver by running the shell script as follows or you can also install driver  in Software&Updates -> Additional Drivers

#!/bin/bash
# Install Nvidia Driver
add-apt-repository ppa:graphics-drivers/ppa
apt-get