Home

Welcome to the TensorNAS wiki!

Installation

Prep

See this article for Tensorflow + CUDA compatibility versions.

To check Tensorflow version run

python3.7 -c 'import tensorflow as tf; print(tf.__version__)' # for Python 3.7

to check CUDA version run

nvcc --version

and to check the cuDNN version run

sudo find /usr -type f -name "cudnn_version.h" -print0 | sudo xargs -0 grep CUDNN_MAJOR -A 2

Arch Linux

Python Packages

yay python-pycuda

CUDA 11.0 + cuDNN 8.0 + Tensorflow 2.4.0

yay cudnn8-cuda11.0

sudo pacman -Sy python-tensorflow