Difference between revisions of "Tensorflow"
KevinYager (talk | contribs) (→Installation instructions) |
KevinYager (talk | contribs) (→Virtualenv method) |
||
Line 15: | Line 15: | ||
##: apt-get update | ##: apt-get update | ||
##: aptitude install cuda | ##: aptitude install cuda | ||
+ | # Install CUDA (alternate method, if above doesn't work): | ||
+ | ## Download CUDA runfile from [https://developer.nvidia.com/cuda-downloads the nVida site] (CUDA toolkit 7.5 runfile): | ||
+ | ##: wget http://developer.download.nvidia.com/compute/cuda/7.5/Prod/local_installers/cuda_7.5.18_linux.run | ||
# Set up environment | # Set up environment | ||
##: nano ~/.bashrc | ##: nano ~/.bashrc |
Revision as of 13:56, 13 September 2016
TensorFlow is an open source library for deep learning.
Installation instructions
Virtualenv method
Detailed instructions are provided on the TensorFlow website.
- Install CUDA
- Download CUDA from the nVidia site (CUDA toolkit 7.5):
- Update root path (if necessary):
- sudo nano /root/.bashrc
- export PATH=$PATH:/usr/local/sbin:/usr/sbin:/sbin:/opt/conda/bin
- Install deb (c.f. instructions here):
- sudo -s
- dpkg -i cuda-repo-ubuntu1504-7-5-local_7.5-18_amd64.deb
- apt-get update
- aptitude install cuda
- Install CUDA (alternate method, if above doesn't work):
- Download CUDA runfile from the nVida site (CUDA toolkit 7.5 runfile):
- Set up environment
- nano ~/.bashrc
- # Add these lines to the end of your .bashrc file:
- export CUDA_HOME=/usr/local/cuda-7.5
- export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
- PATH=${CUDA_HOME}/bin:${PATH}
- export PATH
- Create Virtualenv:
- virtualenv --system-site-packages ~/tensorflow
- Activate Virtualenv:
- source ~/tensorflow/bin/activate
- # Prompt should change to start with "(tensorflow)$"
- Install TensorFlow into Virtualenv:
- # Ubuntu/Linux 64-bit, GPU enabled, Python 2.7
- # Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.
- (tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.10.0-cp27-none-linux_x86_64.whl
- (tensorflow)$ pip install --upgrade $TF_BINARY_URL
- Test:
- (tensorflow)$ python
- >>> import tensorflow as tf