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
附截图:
验证成功的截图:
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版本的环境配置与安装教程