I am setting TensorFlow on:
- Ubuntu 16.04 LTS 64-bit
- 16 GiB RAM
- AMD Athlon(tm) II X4 640 Processor × 4
- GeForce GTX 1050 Ti/PCIe/SSE2
Check if you have NVidia CUDA GPU
uki@uki-p6710f:~$ lspci | grep -i nvidia
01:00.0 VGA compatible controller: NVIDIA Corporation Device 1c82 (rev a1)
01:00.1 Audio device: NVIDIA Corporation Device 0fb9 (rev a1)
Check the name of your OS
uki@uki-p6710f:~$ uname -m && cat /etc/*release
x86_64
DISTRIB_ID=Ubuntu
DISTRIB_RELEASE=16.04
DISTRIB_CODENAME=xenial
DISTRIB_DESCRIPTION="Ubuntu 16.04.1 LTS"
NAME="Ubuntu"
VERSION="16.04.1 LTS (Xenial Xerus)"
ID=ubuntu
ID_LIKE=debian
PRETTY_NAME="Ubuntu 16.04.1 LTS"
VERSION_ID="16.04"
HOME_URL="http://www.ubuntu.com/"
SUPPORT_URL="http://help.ubuntu.com/"
BUG_REPORT_URL="http://bugs.launchpad.net/ubuntu/"
VERSION_CODENAME=xenial
UBUNTU_CODENAME=xenial
Check C compiler
uki@uki-p6710f:~$ gcc --version
gcc (Ubuntu 5.4.0-6ubuntu1~16.04.4) 5.4.0 20160609
Copyright (C) 2015 Free Software Foundation, Inc.
This is free software; see the source for copying conditions. There is NO
warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
Ubuntu Headers
uki@uki-p6710f:~$ sudo apt-get install linux-headers-$(uname -r)
[sudo] password for uki:
Reading package lists... Done
Building dependency tree
Reading state information... Done
linux-headers-4.4.0-62-generic is already the newest version (4.4.0-62.83).
linux-headers-4.4.0-62-generic set to manually installed.
The following packages were automatically installed and are no longer required:
linux-headers-4.4.0-31 linux-headers-4.4.0-31-generic linux-image-4.4.0-31-generic linux-image-extra-4.4.0-31-generic
Use 'sudo apt autoremove' to remove them.
0 upgraded, 0 newly installed, 0 to remove and 89 not upgraded.
Download newest CUDA installer (1.4GB)
https://developer.nvidia.com/cuda-downloads
https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda_8.0.61_375.26_linux-run
Execute CUDA installer
cd ~/Downloads/uki@uki-p6710f:~/Downloads$ ls -alt
Mar 5 14:54 cuda_8.0.61_375.26_linux.run
Mar 5 14:54 cuda_8.0.61_375.26_linux.run
Feb 2 09:53 NVIDIA-Linux-x86_64-375.10.run
Set environment variables
uki@uki-p6710f:~$ nano ~/.bashrcexport PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64
uki@uki-p6710f:~$ echo $LD_LIBRARY_PATH
/usr/local/cuda-8.0/lib64
Install TensorFlow via pip3 (Python 3.5)
uki@uki-p6710f:~$ python --version
Python 3.5.2 :: Anaconda 4.3.0 (64-bit)
$ sudo apt install python3-pip
$ conda info --envs
# conda environments:
#
tensorflow /home/uki/anaconda3/envs/tensorflow
root * /home/uki/anaconda3
# conda environments:
#
tensorflow /home/uki/anaconda3/envs/tensorflow
root * /home/uki/anaconda3
$ conda env create -f /Users/ukilucas/dev/uki.guru/conda_enviroment_GPU.yml
$source activate tensorflow
Setting Jupyter kernel to match Python conda environment
http://ukitech.blogspot.com/2017/02/kernel.html