TensorFlow: Difference between revisions

 
(2 intermediate revisions 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>
Line 23: Line 18:


===Install TF1===
===Install TF1===
The last official version of TensorFlow v1 is 1.15. This version does not work on RTX 3000+ GPUs. Your code will run but output bad results.<br>
The last official version of TensorFlow v1 is 1.15. This version does not work on RTX 3000+ (Ampere) GPUs. Your code will run but output bad results.<br>
If you need TensorFlow v1, see [https://github.com/NVIDIA/tensorflow nvidia-tensorflow].
If you need TensorFlow v1, see [https://github.com/NVIDIA/tensorflow nvidia-tensorflow].
<pre>
<pre>