TensorFlow: Difference between revisions

TensorFlow (view source)

115 bytes removed , 28 January 2021

5,337

edits

@@ Line 10: / Line 10: @@
 Install tensorflow and [https://www.tensorflow.org/addons/overview tensorflow-addons]
 <pre>
-# Install cuda and cudnn if necessary
+conda install tensorflow-gpu
-conda install cudatoolkit=11.0.221
+pip install tensorflow-addons
-pip install tensorflow tensorflow-addons
 </pre>
-* Run <code>conda search cudatoolkit</code> to see other versions of cuda available
+* If you prefer, you can install only cuda and cudnn from conda:
 ** See [https://www.tensorflow.org/install/source#linux https://www.tensorflow.org/install/source#linux] for a list of compatible Cuda and Cudnn versions.
-* On Windows, there is no cudnn available for cuda 11 in conda's repos. You will need to install this manually by downloading [https://developer.nvidia.com/cuDNN cudnn] and copying the binaries to the environment's <code>Library/bin/</code> directory.
+** <code>conda search cudatoolkit</code> to which versions of cuda available
+** Download [https://developer.nvidia.com/cuDNN cudnn] and copy the binaries to the environment's <code>Library/bin/</code> directory.
 ===Install TF1===