我在笔记本上一开始跑cpu版本的tensorflow,感觉太慢了,跑个posenet卡成狗,就换成gpu版的。
记录下安装步骤
我的笔记本win10系统64位,mx250显卡。IDE是用的是pycharm https://www.jetbrains.com/pycharm/ 自己下
python3.6.8 cuda 9.1 cudnn7.1 tensorflow版本是1.8.0
python3.6.8可以自己去官网下载 https://www.python.org/downloads/windows/
下第二个就行了,下好了安装下。
cuda cudnn 还有tensorflow可能比较慢
cuda https://developer.nvidia.com/cuda-toolkit-archive
cudnn https://developer.nvidia.com/cudnn 这个要注册才能下载
tensorflow 因为你用pip装的可能不支持avx2啥的 https://github.com/fo40225/tensorflow-windows-wheel 在这边找个对应版本的
上面3个可能下载有点慢 我将他们放在百度盘里
链接: https://pan.baidu.com/s/1tVljecDtQYBd3XbcExaTsw 提取码: 2nr2
先把cuda装好,先运行cuda_9.1.85_win10 然后那三个1 2 3依次装下。
把cudnn9.1压缩包解压
将bin include lib三个文件夹复制到你装的cuda那个下面
一般在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1 下面
然后环境变量 系统变量Path里添加
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\libnvvp
下面装gpu版tensorflow 一般cmd里打命令 pip install tensorflow-gpu 就行,但这边安装的可能不支持avx2指令啥的
这边用下载了whl文件安装
将那个tensorflow文件复制到d盘根目录下
在cmd里输入 d:
pip install xxx (xxx表示那个whl文件名称)
然后就行了
可以测试下
cmd里输入命令
python
import tensorflow as tf
tf.__version__
OK
来源:CSDN
作者:Isaac320
链接:https://blog.csdn.net/Isaac320/article/details/103609257