TensorFlow: Difference between revisions

 
(One intermediate revision by the same user not shown)
Line 2: Line 2:


==Install==
==Install==
* Install CUDA and CuDNN
* Create a conda environment with python 3.5+
** <code>conda create -n my_env python=3.8</code>
* Install with pip


===Install TF2===
===Install TF2===
Easiest way is to install using conda to get a compatible tensorflow, cuda, and cudnn installed together.
See https://www.tensorflow.org/install/pip
Install tensorflow and [https://www.tensorflow.org/addons/overview tensorflow-addons]
Install tensorflow and [https://www.tensorflow.org/addons/overview tensorflow-addons]
<pre>
<pre>
conda install tensorflow-gpu
pip install tensorflow-addons
pip install tensorflow-addons
</pre>
</pre>