(Tensorflow学习) Win10+mx250 tensorflow gpu版安装

自古美人都是妖i 提交于 2019-12-20 02:20:55

我在笔记本上一开始跑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

 

 

 

 

 

 

  

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!