Difference between revisions of "Tensorflow"
KevinYager (talk | contribs) (→Virtualenv method) |
KevinYager (talk | contribs) (→Installation instructions) |
||
Line 22: | Line 22: | ||
##:: blacklist nouveau | ##:: blacklist nouveau | ||
##:: options nouveau modeset=0 | ##:: options nouveau modeset=0 | ||
+ | ##: sudo -s | ||
+ | ##: update-initramfs -u | ||
+ | ## Install | ||
+ | ##: chmod +x cuda_7.5.18_linux.run | ||
+ | ##: sudo ./cuda_7.5.18_linux.run | ||
# Set up environment | # Set up environment | ||
##: nano ~/.bashrc | ##: nano ~/.bashrc |
Revision as of 14:11, 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):
- Disable nouveau:
- sudo nano /etc/modprobe.d/blacklist-nouveau.conf
- blacklist nouveau
- options nouveau modeset=0
- sudo -s
- update-initramfs -u
- sudo nano /etc/modprobe.d/blacklist-nouveau.conf
- Install
- chmod +x cuda_7.5.18_linux.run
- sudo ./cuda_7.5.18_linux.run
- 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
- nano ~/.bashrc
- 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