programming_languages:python:tensorflow

Tensorflow

Installation

List installed packages

apt list --installed

Find out CUDA version:

nvcc --version
cat /usr/local/cuda/version.txt

Find out cuDNN version:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
sudo apt-get install cuda-libraries-9-0

Tensorflow prebuilts CUDA 9.0 Prebuild tf binaries are using cuDNN 7.2, TensorRT 4 since v. 1.11.0: https://github.com/tensorflow/tensorflow/releases

  • programming_languages/python/tensorflow.txt
  • Last modified: 2018/09/29 17:57
  • by phreazer