Windows安裝tensorflow經驗總結(尤其安裝GPU版本的細看)


我的安裝順序:先Anaconda,然后CUDA和CUDAnn,最后在虛擬環境下安裝tensorflow

anaconda下載鏈接https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

CUDA下載鏈接 https://developer.nvidia.com/cuda-toolkit-archive

cudann下載鏈接https://developer.nvidia.com/rdp/cudnn-archive

1、  如果采用anaconda進行安裝,切記要把之前安裝的python卸載干凈。

2、  如果是安裝GPU版本tensorflow一定要查看自己電腦支持的CUDA版本,選擇CUDA與CUDAnn以及tensorlow版本也要對應

3、  安裝CUDA時若勾選安裝Visual Studio Integration出現失敗,安裝時可不勾選該選項,后續可以通過下面的方式單獨實現安裝

4、  在anaconda下安裝tensorflow首先要建立tensorflow的虛擬環境,然后在虛擬環境下安裝tensorflow

5、  如果自己電腦的CPU版本過低,切記不要安裝tensorflow2.0版本以上,也就是在安裝時不要直接使用該命令:pip install tensorflow-gpu,而要指定安裝的tensorflow版本

6、  在線下載tensorflow時可能會彈出找不到與之匹配的版本的錯誤,可能是網絡的原因。可以在下載命令后面添加鏡像源

 

2點:右鍵桌面NVIDIA控制面板

 

 

 

點擊左下角的系統信息

點擊組件可以查看CUDA支撐版本,藍色區域所示CUDA10.0.132

 

 

 

3點:單獨配置Visual Studio Integration

先抽取cuda_9.0.176_win10.exe記住抽取路徑

找到MSBuildExtensions文件夾

E:\study\CUDA_package\CUDAVisualStudioIntegration\extras\visual_studio_integration\MSBuildExtensions

將該文件夾下所有文件拷貝到C:\Program Files (x86)\MSBuild\Microsoft.Cpp\v4.0\V140\BuildCustomizations”文件夾中

找到CudaProjectVsWizards文件夾

E:\study\CUDA_package\CUDAVisualStudioIntegration\extras\visual_studio_integration\CudaProjectVsWizards

將整個文件夾分別拷貝到E:\study\VS2015\Common7\IDE\Extensions 以及E:\study \VS2015\Common7\IDE\CommonExtension下,已自己安裝VS的路徑為准

 

然后找到VS開發人員命令提示符,以管理員身份運行下面這條命令

devenv.com /setup /nosetupvstemplates

 

 

再打開VS,即可看到項目模板里的NVIDIA/CUDA項目選擇

 

 

第4點:anaconda安裝成功后找到anaconda Prompt,點擊運行

在Prompt運行下面語句:創建虛擬環境

conda create --name tensorflow python=3.5

然后輸入activate tensorflow激活虛擬環境;

激活的標志是命令前會有(tensorlow)如下圖所示:

 在虛擬環境下下載安裝tensorflow,

 

 

第5點:指定版本tensorflow安裝

pip install tensorflow-gpu==1.8.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/

(后面鏡像源可以自己更換)

tensorflow版本選擇參考本網址,與CUDA版本匹配,cpu不行的適當往低版本靠https://tensorflow.google.cn/install/source_windows

 

 

測試步驟(tensorflow是否安裝成功)

import tensorflow as tf

sess = tf.Session()

a = tf.constant(10)

b = tf.constant(20)

sess.run(a+b)

 


免責聲明!

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



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