5,337
edits
TensorFlow: Difference between revisions
→Install TF1
| Line 23: | Line 23: | ||
<pre> | <pre> | ||
# Install compatible cuda and cudnn versions. | |||
conda install cudatoolkit=10.0.130 cudnn=7.6.5 | conda install cudatoolkit=10.0.130 cudnn=7.6.5 | ||
# Install tensorflow | |||
pip install tensorflow-gpu==1.15 | pip install tensorflow-gpu==1.15 | ||
# Test GPU support | |||
python -c "import tensorflow as tf;print(tf.test.is_gpu_available())" | |||
</pre> | </pre> | ||