tensorflow-gpu在win10下的安裝

參考：https://blog.csdn.net/gyp2448565528/article/details/79451212

按照原博主的方法在本身的機器上會有一點小錯誤，下面的方法略有不一樣

 

環境：win10 64位系統，帶nVidia顯卡

在https://www.geforce.com/hardware/technology/cuda/supported-gpus查看是否支持本身的顯卡用於加速訓練

學tensorflow以前試過一次安裝GPU版本的TF，網上有不少教程，看着都挺繁瑣，其實直接在anaconda裏配置一下就行了，但anaconda默認下載源下載速度極慢，GPU版本的下載的時候老是出現錯誤：multiple error，百度了一下沒找到錯誤緣由，應該是下載速度過慢致使的相似time out 的錯誤，做罷。如今用CPU版的TF訓練網絡速度實在不能忍，因而昨晚按照上面貼出的博文的方法，折騰了一番，最終成功安裝，如今捋一捋，步驟以下：

 

1. 下載安裝anaconda，安裝時注意勾選add anaconda to my PATH environment variable

 

2. 打開cmd,輸入如下命令：

conda create -n tensorflow pip python=3.6

遇到y/n時都選擇y（個人python版本是3.6的，因此使用3.6的版本）

 

3. 輸入命令：

activate tensorflow

 

4. 因爲默認源下載速度極慢，安裝圖中會出現timeout錯誤，遂改用國內的源，採用pip安裝，一開始按照原博主的方法輸入這個命令：

後面的-i多是-index-url的縮寫（我猜的），安裝其餘包時也能夠加入這句來指定爲下載源，然而過了幾分鐘依然彈出timeout錯誤，又查了一下解決辦法，加了一句--default-timeout=100，修改默認等待時間，即輸入下面的命令：

--default-timeout=100 pip install --ignore-installed --upgrade tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

pip install--ignore-installed --upgrade tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

等待一段時間後，終於完成安裝，以下圖：


接着在命令行輸入python，再輸入import tensorflow，提示沒安裝cuda：

（圖片引用自上面貼出的博文）

按照原文中提供的cuda和cudnn下載並安裝，並在 「個人電腦-管理-高級設置-環境變量」 中找到path，添加如下環境變量（cuda使用默認安裝路徑）：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include

 

 按照原文的方法，添加環境變量後輸入命令：

import tensorflow

然而仍是出現上面沒有在path路徑中找到dll文件的錯誤，因而輸入下面的命令，打印當前的變量：

echo %path%

發現上面的4個環境變量並無添加進去，原來還須要重啓一下，重啓以後再輸入上面的命令，發現4條環境變量已成功添加

再在命令行裏輸入：

activate tensorflow

再輸入：

python

再輸入：

import tensorflow

終於沒再彈出錯誤：

 

 

---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------

注意，上面安裝的tf在anaconda navigator裏建立了一個名爲tensorflow的環境，但在這個環境裏是找不到tensorflow的包的，由於這個tf是經過pip安裝的，因此不會在裏面顯示，但這個包的確被安裝到anaconda的這個新環境的路徑裏了，只是不能用navigator來管理

 

在pycharm中的配置：

file--settings--project interpreter，單擊右邊的齒輪圖標，選擇add local，添加本地解釋器，如圖：

 

切換到existing environment，找到剛安裝的tensorflow環境裏的python.exe，單擊OK:

 

 project就切換到含有tensorflow-gpu包的新環境裏：

 

等等。。。。爲何package裏找不到tensorflow？原來是使用了conda package manager，把它點掉，這樣就能正常在PC裏使用TF啦：

 

之後若是想在這個環境裏安裝新的包，應該是在命令行裏activate tensorflow，而後再pip install XXX。。。