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

我在笔记本上一开始跑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/1Fbi0NRx5cJZuUCJ6g_UmBw 
提取码:ui6e 

先把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

 

 

 

 

 

 

  

评论 14
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值