TensorFlow: Difference between revisions
No edit summary |
|||
Line 1: | Line 1: | ||
TensorFlow is the famous machine learning library by Google | TensorFlow is the famous machine learning library by Google | ||
==Install== | |||
* Install CUDA and CuDNN | |||
* Create a conda environment with python 3.7 | |||
** You can also just create a tensorflow environment using conda | |||
** <code>conda create -n my_env tensorflow | |||
* Install with conda or pip | |||
===Install TF1=== | |||
<pre> | |||
conda install tensorflow | |||
</pre> | |||
===Install TF2=== | |||
<pre> | |||
conda install tensorflow-gpu==1.15 | |||
</pre> | |||