Installation
A note for installing tensorflow under ubuntu 16.10 with nvidia GTX-750 Ti graphic card
Install the following packages
sudo apt install python-pip python-dev libcupti-dev libcublas8.0 nvidia-cuda-dev nvidia-cuda-toolkit
Install tensorflow GPU version
pip install --upgrade tensorflow-gpu
Download cudnn from
NVIDIA website (you need to register a account before download it)
Extract the file and execute below commands
sudo cp -P include/cudnn.h /usr/include
sudo cp -P lib64/libcudnn* /usr/lib/x86_64-linux-gnu/
sudo chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn*
Reference
TensorFlow official website for installation
Nvidia officiail website for installation
No comments:
Post a Comment