[學習筆記] Tensorflow安裝配置

摘要

對於涉及驗證碼的反爬,比較有效的方式就是利用機器學習訓練模型,破解驗證碼,以前有零散的接觸一些機器學習相關知識,但記憶過於混亂因而邊學邊記錄學習遇到的坑python

安裝
  • 使用國內鏡像進行加速下載tensorflow
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow
  • 下載Cuda Cudnn

Nvidia下載速度比較慢,建議使用代理,注意版本兼容性,環境變量配置如圖shell

13b731552f423d916cc121faf8b2269.png

  • 測試可否使用Gpu加速
import tensorflow as tf
import os

os.environ['TF_CPP_MIN_LOG_LEVEL']='2'

a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)

print('GPU:', tf.test.is_gpu_available())
  • 異常處理

按照以上步驟安裝,我出現了Could not load dynamic library 'cudart64\_101.dll 異常,我按照參考一文章提示解決了機器學習

[1] 解決tensorflow-gpu 2.1出現錯誤「Could not load dynamic library 'cudart64_101.dll'學習

相關文章
相關標籤/搜索