5,337
edits
Line 37: | Line 37: | ||
python -c "import tensorflow as tf;print(tf.test.is_gpu_available())" | python -c "import tensorflow as tf;print(tf.test.is_gpu_available())" | ||
</pre> | </pre> | ||
;Notes | |||
* 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] | |||
==Usage (TF2)== | ==Usage (TF2)== |