win10下安装TensorFlow2.0-gpu踩过的坑

1. CUDA的坑:

版本:10.0
一般需要更新一下驱动程序,下载个“360驱动大师”可简单更新完成。注:驱动版本要410.x以上。
假若使用过程中发现没有调用到GPU,则可能是下面这篇博客说到的问题:
解决TensorFlow2.0不能用GPU运算的问题

2. 下载TensorFlow2.0-gpu

个人使用的是:

conda create -n tf2_GPU tensorflow-gpu

更改为:

pip install -i https://pypi.doubanio.com/simple tensorflow-gpu==2.0.0

3. 关于验证TensorFlow-gpu是否安装成功:

验证是否安装成功代码:

import tensorflow as tf 
print(tf.__version__)  #输出版本号
print(tf.test.is_gpu_available())  #安装成功应该输出True

进入anaconda的相应环境后:验证方式
a.输入python进入交互环境验证一般没什么问题。
b.输入ipython进入交换环境则需要看看tensorflow-GPU环境中有没有安装好ipython,输入命令where ipython查看。
不然输入print(tf.test.is_gpu_available())输出为Flase
对比一下:确保该环境下是否有ipython
附截图:
确保该环境下有ipython
验证成功的截图:
在这里插入图片描述

4. 关于AttributeError: module 'tensorflow' has no attribute 'Session'问题:

验证代码时,若输入:

import tensorflow as tf
sess=tf.Session(config=tf.ConfigProto(log_device_placement=True))

会出现报错:AttributeError: module 'tensorflow' has no attribute 'Session'
在这里插入图片描述原因是TensorFlow的版本问题。应该为:

import tensorflow as tf
sess=tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(log_device_placement=True))

参考博客:AttributeError: module ‘tensorflow’ has no attribute ‘Session’

附安装过程参考的博客:
更新系类:更新
Windows 10 下 Anaconda 轻松安装 Tensorflow 2.0 GPU 版本
完整TensorFlow2.0.0-gpu(正式版) + Anaconda + Win10 安装教程
Win10安装CUDA10和cuDNN
CUDA、CUDNN在windows下的安装及配置
tensorflow2.0GPU版本的环境配置与安装教程

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值