煎熬的趟坑之旅就不多談了,下面是準備工作:

  1. 準備一個梯子,這點很重要,在後面安裝過程中,如果使用國內鏡像只能安裝1.1.0版本的tensorflow-gpu。
  2. 安裝最新版anaconda,當前是python3.7版本。
  3. 確認你的顯卡是N卡並能夠支持Cuda,如果從這個網址能找到就說明支持(後面跟著的Cuda版本是最低支持版本)。
  4. 更新顯卡驅動,由於我們等會要安裝最新版本的Cuda runtime version,如果顯卡的Cuda driver version落後的話會報錯。從這個網站下載最新驅動並安裝即可。

Step1:

google搜索use anaconda to install tensorflow,找到anaconda官網的tensorflow安裝方法,當前結果如下:

從開始菜單找到Anaconda Prompt,打開,輸入以上命令,直接開始安裝tensorflow-gpu的各種依賴,我們可以看到下載的依賴裡面已經包含了cuda和cudnn,當前安裝的版本是tensorflow-gpu-1.13+cuda10+cudnn7.3。

安裝過程大概持續一個小時。如果有的包下載失敗重新執行一遍上個命令就能繼續安裝。

Step2:

鍵入:

activate tensorflow_gpuenv
python #進入python
#下面測試tensorflow-gpu是否安裝成功
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

成功截圖如下:


推薦閱讀:
相关文章