5,337
edits
Line 11: | Line 11: | ||
Install tensorflow and [https://www.tensorflow.org/addons/overview tensorflow-addons] | Install tensorflow and [https://www.tensorflow.org/addons/overview tensorflow-addons] | ||
<pre> | <pre> | ||
# Install cuda and cudnn if necessary | |||
conda install cudatoolkit=10.1.243 cudnn=7.6.5 | |||
pip install tensorflow tensorflow-addons | pip install tensorflow tensorflow-addons | ||
</pre> | </pre> |