5,337
edits
TensorFlow: Difference between revisions
→Install TF2
| Line 11: | Line 11: | ||
<pre> | <pre> | ||
# Install cuda and cudnn if necessary | # Install cuda and cudnn if necessary | ||
conda install cudatoolkit= | conda install cudatoolkit=11.0.221 | ||
pip install tensorflow tensorflow-addons | pip install tensorflow tensorflow-addons | ||
</pre> | </pre> | ||
* Run <code>conda search cudatoolkit</code> to see other versions of cuda available | |||
* 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. | |||
===Install TF1=== | ===Install TF1=== | ||