windows10安装tensorflow的gpu版本(pip3安装方式)

前言:

   TensorFlow 有cpu和 gpu两个版本:gpu版本需要英伟达CUDA 和 cuDNN 的支持,cpu版本不需要;本文主要安装gpu版本。

1、环境

  • gpu:确认你的显卡支持 CUDA,这里确认
  • vs2015运行时库:下载64位的,这里下载,下载后安装。
  • python 3.6/3.5:下载64位的,这里下载,下载后安装。
  • pip 9.0.1(确认pip版本 >= 8.1,用pip -V 查看当前 pip 版本,用python -m pip install -U pip升级pip 。)

注意:pip其实在安装python的时候就自带了,但是需要将其(XXX:\ Python\Python36\Scripts)配置到环境变量(path)中,红色字体代表python安装 的路径。

2、安装

(1)、下载 CUDA 
    Google 官方推荐CUDA8 ,可以去官网下载:cuda

  • 安装CUDA 

   直接双击exe安装,按提示下一步装到底就行。装完测试一下是否安装成功,命令行输入 nvcc -V ,看到版本信息就表示安装成功了。

(2)、下载cuDNN

 tensorflow版本不同,需要的cuDNN版本也不一样,具体看tensorflow的发布说明,如:tensorflow1.3的发布说明

  • 配置cuDNN

   下载到对应版本的cuDNN(tensorflow1.3需要cuDNN6,可到 https://www.zhihu.com/question/37082272下载)后,解压:

  1. 将解压后的bin目录配置到系统环境变量(path)中;
  2. 将相应的文件拷贝到相应的目录:  

  copy cudnn\bin\cudnn64_5.dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin\
    copy cudnn\include\cudnn.h to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include\
    copy cudnn\lib\x64\cudnn.lib to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64\

(3)、安装tensorflow

[html] view plain copy

  1. # GPU版本
  2. pip3 install --upgrade tensorflow-gpu
  3. # CPU版本
  4. pip3 install --upgrade tensorflow

(4)、卸载tensorflow

[html] view plain copy

  1. pip3 uninstall tensorflow

3、测试

[html] view plain copy

  1. import tensorflow as tf
  2. a = tf.random_normal((100, 100))
  3. b = tf.random_normal((100, 500))
  4. c = tf.matmul(a, b)
  5. sess = tf.InteractiveSession()
  6. sess.run(c)

issue:

[html] view plain copy

  1. Traceback (most recent call last):
  2. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 18, in swig_import_helper
  3. return importlib.import_module(mname)
  4. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\importlib\__init__.py", line 126, in import_module
  5. return _bootstrap._gcd_import(name[level:], package, level)
  6. File "<frozen importlib._bootstrap>", line 986, in _gcd_import
  7. File "<frozen importlib._bootstrap>", line 969, in _find_and_load
  8. File "<frozen importlib._bootstrap>", line 958, in _find_and_load_unlocked
  9. File "<frozen importlib._bootstrap>", line 666, in _load_unlocked
  10. File "<frozen importlib._bootstrap>", line 577, in module_from_spec
  11. File "<frozen importlib._bootstrap_external>", line 906, in create_module
  12. File "<frozen importlib._bootstrap>", line 222, in _call_with_frames_removed
  13. ImportError: DLL load failed: The specified module could not be found.
  14. During handling of the above exception, another exception occurred:
  15. Traceback (most recent call last):
  16. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 41, in <module>
  17. from tensorflow.python.pywrap_tensorflow_internal import *
  18. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 21, in <module>
  19. _pywrap_tensorflow_internal = swig_import_helper()
  20. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 20, in swig_import_helper
  21. return importlib.import_module(‘_pywrap_tensorflow_internal‘)
  22. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\importlib\__init__.py", line 126, in import_module
  23. return _bootstrap._gcd_import(name[level:], package, level)
  24. ImportError: No module named ‘_pywrap_tensorflow_internal‘
  25. During handling of the above exception, another exception occurred:
  26. Traceback (most recent call last):
  27. File "<pyshell#0>", line 1, in <module>
  28. import tensorflow as tf
  29. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
  30. from tensorflow.python import *
  31. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\__init__.py", line 51, in <module>
  32. from tensorflow.python import pywrap_tensorflow
  33. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 52, in <module>
  34. raise ImportError(msg)
  35. ImportError: Traceback (most recent call last):
  36. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 18, in swig_import_helper
  37. return importlib.import_module(mname)
  38. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\importlib\__init__.py", line 126, in import_module
  39. return _bootstrap._gcd_import(name[level:], package, level)
  40. File "<frozen importlib._bootstrap>", line 986, in _gcd_import
  41. File "<frozen importlib._bootstrap>", line 969, in _find_and_load
  42. File "<frozen importlib._bootstrap>", line 958, in _find_and_load_unlocked
  43. File "<frozen importlib._bootstrap>", line 666, in _load_unlocked
  44. File "<frozen importlib._bootstrap>", line 577, in module_from_spec
  45. File "<frozen importlib._bootstrap_external>", line 906, in create_module
  46. File "<frozen importlib._bootstrap>", line 222, in _call_with_frames_removed
  47. ImportError: DLL load failed: The specified module could not be found.
  48. During handling of the above exception, another exception occurred:
  49. Traceback (most recent call last):
  50. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 41, in <module>
  51. from tensorflow.python.pywrap_tensorflow_internal import *
  52. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 21, in <module>
  53. _pywrap_tensorflow_internal = swig_import_helper()
  54. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 20, in swig_import_helper
  55. return importlib.import_module(‘_pywrap_tensorflow_internal‘)
  56. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\importlib\__init__.py", line 126, in import_module
  57. return _bootstrap._gcd_import(name[level:], package, level)
  58. ImportError: No module named ‘_pywrap_tensorflow_internal‘
  59. Failed to load the native TensorFlow runtime.
  60. See https://www.tensorflow.org/install/install_sources#common_installation_problems
  61. for some common reasons and solutions.  Include the entire stack trace
  62. above this error message when asking for help.

如果在 import tensorflow 的时候出现此问题,很有可能是没有装相应的vs运行时库,如tensorflow1.3,需要vs2015的运行时库即Visual C++ Redistributable for Visual Studio 2015

时间: 2024-08-28 20:06:49

windows10安装tensorflow的gpu版本(pip3安装方式)的相关文章

linux 安装tensorflow(gpu版本)

一.安装cuda 具体安装过程见我的另一篇博客,ubuntu16.04下安装配置深度学习环境 二.安装tensorflow 1.具体安装过程官网其实写的比较详细,总结一下的话可以分为两种:安装release版本和源码编译安装.因为源码编译安装比较繁琐,且需要安装谷歌自己的编译器bazel,所以我选择安装编译好的. 2.我写这篇博客的时候tensorflow更新到了1.4.0,安装编译好的一定看版本,因为每个版本依赖的底层库是不一样的. 1.4.0版本安装之前需要安装CUDA-8,cuDNN v6

Win10安装Tensorflow的Gpu版本问题

Cuda,Cudnn 和 Tensorflow gpu这三个东西要兼容 先下载最新的Cudnn,和Cuda的兼容性在下载页上有(需要注册): Cudnn下载地址 下载安装Cudnn的对应Cuda版本,然后将Cudnn解压拷贝到Cuda安装目录: Cuda下载地址 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\[VER.] 上面两个都下载最新相兼容的版本,然后Tensorflow gpu也更新为最新版一般就没问题了: pip install

TensorFlow GPU版本的安装与调试

笔者采用python3.6.7+TensorFlow1.12.0+CUDA10.0+CUDNN7.3.1构建环境 PC端配置为GTX 1050+Intel i7 7700HQ 4核心8线程@2.8GHZ TensorFlow-gpu的安装经历实在是坎坷的很 首先显卡一定要支持 没想到的是GTX 1050TI,GTX 1070TI等主流显卡竟然都不支持 (还好我买的是GTX 1050) (并没有暗示需要一块TESLA) 点这里查看CUDA支持列表 其次需要对好版本号,不同的TensorFlow版本

安装caffe-ssd的GPU版本时候报错:BatchReindexLayerTest/2.TestGradient,where TypeParam=caffe::GPUdevice&lt;float&gt;

报错原因: 配置了多显卡环境,如果使用了Intel MKL作为BLAS,可能是Intel MKL的浮点数计算功能没有设置正确,可以使用安装CUDA时运行./deviceQuery时标注的0号GPU("Device 0")跑跑测试试试看 解决方法: export CUDA_VISIBLE_DEVICES=0 然后make runtest 原文地址:https://www.cnblogs.com/laowangxieboke/p/10346853.html

Windows 10 Tensorflow 2 gpu正式版安装和更新日志

Tensorflow 2.0.0 released on2019年10月1日星期二 Link: https://github.com/tensorflow/tensorflow/releases/tag/v2.0.0 本日志是win 10下tf2.0.0正式版的重新安装/更新的精确技术文档. Steps as follows: Step 1: enter into tf installing folder C:\Program Files (x86)\Microsoft Visual Studi

安装caffe-ssd的GPU版本时候报错:BatchReindexLayerTest/2.TestGradient,where TypeParam=caffe::GPUdevice&lt;double&gt;

报错原因:使用CUDA 9.1加双显卡 解决方法:直接将BatchReindexLayerTest的forWord和backWord改为CPU执行 具体来说就是将-/caffe/src/caffe/layers/batch_reindex_layer.cu文件内的 原文地址:https://www.cnblogs.com/laowangxieboke/p/10346869.html

用pip安装tensorflow换成国内源快速安装

Tensorflow  1.X 安装: pip install --index-url https://pypi.douban.com/simple tensorflow 或pip install --index-url http://mirrors.aliyun.com/pypi/simple/ tensorflow Tensorflow  2.X 安装: (豆瓣源,  注意要是  https,  有个s ) pip install -i https://pypi.doubanio.com/s

tensorflow的gpu版本错误

出现错误:E tensorflow/stream_executor/cuda/cuda_event.cc:48] Error polling for event status: failed to query event: CUDA_ERROR_LAUNCH_FAILED: unspecified launch failure 最后发现是gpu算力不足,将网络的节点个数降低,最后顺利运行. 原文地址:https://www.cnblogs.com/jianglinliu/p/10531074.h

Win10下用Anaconda安装TensorFlow

笔者之前在学习TensorFlow,也在自己的笔记本上完成了安装,在PyCharm中进行学习.但是最近为了使用Python的科学计算环境,我把之前的环境卸载了,并用Anaconda重新安装了TensorFlow,这里介绍一下cpu版本的安装方法. 前提检查: 在 https://developer.nvidia.com/cuda-gpus 确认你的显卡支持 CUDA,如果你安装的是gpu版本的tensorflow才需要检查这一步. 确保你的Python版本是3.5 64位. 确保你有稳定的网络连