(傻瓜版)Tensorflow-GPU安裝及TensorflowObjectionDetectionAPI環境搭建

肖昇平發表於2020-10-14

(傻瓜版)Tensorflow-GPU安裝及Tensorflow Objection Detection API環境搭建

搞了好長時間一直出現各種問題,不是版本不對應就是視訊記憶體不夠,好在發現了一種非常香的安裝方式。之前安裝各種依賴包都是使用pip方式,後來發現各種版本問題(cudaToolkit跟cudnn)。這次所有依賴包都是使用conda 下載,conda會自動選定正確版本的依賴包

一、 Tensorflow-GPU安裝

先說一下我的電腦DELL-G3 GTX1650Ti,tensorflow-gpu==1.1.0。

  1. 安裝Anaconda3,安裝時選擇自動新增環境變數。

  2. 開啟Anaconda Prompt箭頭所指

  3. 先建立虛擬環境 ,在Anaconda Prompt裡鍵入

conda create --name 這裡寫環境名稱 我的是tensorflow  python=3.6

在這裡插入圖片描述

此次我選的是Python3.6,因為接下來我要安裝tensorflow-gpu==1.1.0,若選擇安裝tensorflow高版本這裡建議選擇Python3.7。

  1. 啟用所建立的虛擬環境
    鍵入 activate tensorflow(這裡的tensorflow是我上面自己起的名,你的可以是tensorflow-gpu或是別的 都是可以的。)若是啟用成功 在行前括號裡的內容會變成虛擬環境名稱(我的是tensorflow)如下圖
    在這裡插入圖片描述

  2. 傻瓜式安裝tensorflow
    鍵入 conda install tensorflow-gpu=1.1.0這裡一定用conda。然後就等吧 這裡下載的很慢甚至會連線失敗,解決方法兩種:1.科學上網 2.改清華源如下(鍵入如下三行程式碼):

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
  1. 驗證是否安裝成功
    進入python(在Anaconda Prompt裡 先啟用虛擬環境,然後鍵入python 後回車 然後鍵入 import tensorflow as tf tf.test.is_gpu_available() )如下圖:
    ![(https://img-blog.csdnimg.cn/20201014131933175.png#pic_center)
    若是返回True 則安裝成功。如下圖:
    在這裡插入圖片描述
二、Tensorflow Objection Detection API環境搭建

過兩天再寫,待續。。。。。。。。

相關文章