GPU setup - k821209/pipelines GitHub Wiki

apt-get install pciutils 
lspci |grep VGA
service lightdm stop

## for NVIDIA Quadro K620
add-apt-repository -y ppa:xorg-edgers/ppa
apt-get update
apt-get install nvidia-346
#reboot 
sudo add-apt-repository -r ppa:xorg-edgers/ppa

## for NVIDIA TITAN 이거 안하고 바로 cuda install step 에서 깔아주는 드라이버를 사용하는게 더낫다. 
add-apt-repository ppa:graphics-drivers/ppa
apt-get update
apt-get install nvidia-352 nvidia-settings
# cuda 7.5에서 8로 업그레이드 하기
# http://dhaneshr.net/2016/11/09/upgrading-to-cuda-8-0-on-ubuntu-16-04/

# cuda 8 to 9 
# https://gist.github.com/niderhoff/52c514f20337516500084dd7b17af7e2
# https://park-ju-hyeong.github.io/2018/04/05/CUDA-8.0-to-9.0/

$ wget https://developer.nvidia.com/compute/cuda/9.0/Prod/patches/1/cuda-repo-ubuntu1604-9-0-local-cublas-performance-update_1.0-1_amd64-deb \
    -O cuda-patch1.deb
$ md5sum cuda-patch1.deb
aa849a2b5ae80278ed1298e4e799df65   cuda-patch2
$ wget https://developer.nvidia.com/compute/cuda/9.0/Prod/patches/2/cuda-repo-ubuntu1604-9-0-local-cublas-performance-update-2_1.0-1_amd64-deb \
    -O cuda-patch2.deb
$ md5sum cuda-patch2.deb
104a1f4eda51cfc34c56d17da8059bc6   cuda-patch2
$ sudo dpkg -i cuda-patch1.deb
$ sudo dpkg -i cuda-patch2.deb
$ sudo apt update
$ sudo apt -y upgrade

wget http://developer.download.nvidia.com/compute/cuda/7.5/Prod/local_installers/cuda_7.5.18_linux.run #https://developer.nvidia.com/cuda-75-downloads-archive tar -xvzf cudnn-7.5-linux-x64-v5.1.solitairetheme8 # cudadnn download https://developer.nvidia.com/rdp/cudnn-download sudo cp cuda/include/cudnn.h /usr/local/cuda/include sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* apt-get install python-pip python-dev export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc0-cp27-none-linux_x86_64.whl pip install --ignore-installed --upgrade $TF_BINARY_URL export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64" export CUDA_HOME=/usr/local/cuda

```python
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))