Tensorflow gpu win10环境配置

Tensorflow gpu win10


1.安装cuda 9.0
2.解压cudnn-9.0-windows10-x64-v7.1.zip
复制解压文件目录下 bin,include,lib 三个文件夹到 cuda根目录
(默认在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0)

3.安装Anaconda 3
4.打开Anaconda Prompt,创建环境

conda create -n tf-gpu python=3.6

5.进入环境

conda activate tf-gpu

6.升级pip到最新版,防止稍后的安装时,出现错误

python -m pip install --upgrade pip

7.安装tensorflow-gpu版本 (注:奔腾系列处理器可能需要低版本,tensorflow==1.5,原因:不支持AVX指令集

pip install tensorflow-gpu ==1.11.0

8.验证:在命令行中,进入python,并输入以下代码:

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

输出如下,即为环境配置成功

b’ Hello, TensorFlow

注:1.如果import tensorflow时出现以下的错误,可以检查下处理器是否为奔腾系列,是否支持AVX指令集
换成低版本的tensorflow即可 (如1.5)

ImportError: DLL load failed: 动态链接库(DLL)初始化例程失败。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值