1. 程式人生 > >WIN10 + GTX 1060 + Tensorflow GPU版本安裝記錄

WIN10 + GTX 1060 + Tensorflow GPU版本安裝記錄

最近買了臺本做訓練用,因為一直沒GPU,今天晚上配置一下TensorFlow環境,配置如下:

  • 安裝VS2017

官方下載連結:https://visualstudio.microsoft.com/downloads/?rr=https%3A%2F%2Fwww.baidu.com%2Flink%3Furl%3D7607zWVCfrNBww7i-jC9Idd12bUvZTWfWfQ74KDiDRaVRT9ySKsskOPUcKEByZdTqpRaFQXiNPwQ5-JwXD8nm_%26wd%3D%26eqid%3Dbf2cd077000268f2000000035bf57bdb

百度網盤連結: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