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

and to check CUDA version run

nvcc --version

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