WIN10 + GTX 1060 + Tensorflow GPU版本安裝記錄
阿新 • • 發佈:2018-11-30
最近買了臺本做訓練用,因為一直沒GPU,今天晚上配置一下TensorFlow環境,配置如下:
- 安裝VS2017
百度網盤連結:https://pan.baidu.com/s/1ESJnnxpO2sRey0eE1Am4xA
工作負載選擇如下圖所示:
CUDA版本選擇9.0,對應的CUDNN版本為7.4,下面直接給出百度網盤的下載地址。預設安裝即可
百度網盤下載:https://pan.baidu.com/s/1Ak21bYpIPd-H-sjLQ498Dg
百度網盤下載地址: https://pan.baidu.com/s/1WZkEjTHgTAEKlPNCWDu5GQ
下載解壓後(裡邊有個三個檔案)拷貝到cuda9.0檔案下 ,如下圖所示:
官網下載地址(速度比較慢): https://www.anaconda.com/download/
百度網盤下載地址:https://pan.baidu.com/s/1qDe2mEoFt9cjjQS0B0poDA
安裝需注意將Anaconda新增到PATH環境變數中,勾選第一個複選框。
以管理員身份執行cmd,然後配置清華映象,輸入指令:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
建立執行環境,輸入指令:
conda create -n tensorflow-gpu python=3.6
啟用並進入環境,使後續指令在啟用的環境中生效,輸入指令:
conda activate tensorflow-gpu
升級pip到最新版,防止稍後的安裝時,出現錯誤(筆者初次在安裝tensorflow時,沒有更新pip到最新版,導致下載到一半出現錯誤),輸入指令:
python -m pip install --upgrade pip
安裝TensorFlow GPU版本,輸入命令:
pip install tensorflow-gpu