5,337
edits
TensorFlow: Difference between revisions
→Install TF1
| Line 20: | Line 20: | ||
===Install TF1=== | ===Install TF1=== | ||
Just install using Conda. It's the easiest way. | |||
<pre> | <pre> | ||
conda install tensorflow-gpu=1.15 | |||
conda | |||
</pre> | </pre> | ||
;Notes | ;Notes | ||
* Conda will automatically install a compatible cuda and cudnn into the cuda environment. Your host OS only needs to have a sufficiently new version of nvidia drivers installed. | |||
* Sometimes, I get <code>CUDNN_STATUS_INTERNAL_ERROR</code>. This is fixed by setting the environment variable <code>TF_FORCE_GPU_ALLOW_GROWTH=true</code> in my conda env. See [https://stackoverflow.com/questions/46826497/conda-set-ld-library-path-for-env-only Add env variables to conda env] | * Sometimes, I get <code>CUDNN_STATUS_INTERNAL_ERROR</code>. This is fixed by setting the environment variable <code>TF_FORCE_GPU_ALLOW_GROWTH=true</code> in my conda env. See [https://stackoverflow.com/questions/46826497/conda-set-ld-library-path-for-env-only Add env variables to conda env] | ||