Skip to content

Instantly share code, notes, and snippets.

@rgtjf
Last active April 28, 2018 04:42
Show Gist options
  • Save rgtjf/e83a2d5eb14be7e92d349ebefbc25d4f to your computer and use it in GitHub Desktop.
Save rgtjf/e83a2d5eb14be7e92d349ebefbc25d4f to your computer and use it in GitHub Desktop.

Install Tensorflow>=1.5.0

Install CUDA

Install cuDNN v7

  • cudnn-install

     tar -xzvf cudnn-9.0-linux-x64-v7.tgz
     # cuda -> cuda-9.0
     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*

GPU drivers supporting CUDA Toolkit

F.A.Q

  1. Failed to initialize NVML: Driver/library version mismatch
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment