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安裝,一開始按照原博主的方法輸入這個命令:

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

后面的-i可能是-index-url的縮寫(我猜的),安裝其他包時也可以加入這句來指定為下載源,然而過了幾分鍾依然彈出timeout錯誤,又查了一下解決辦法,加了一句--default-timeout=100,修改默認等待時間,即輸入下面的命令:

pip install --default-timeout=100 --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。。。


 
        

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM