TensorFlow安装(2.1.0版)

TensorFlow安装(2.1.0版)及问题解决方法

主要流程参照这位大佬的:博客如下:

https://blog.csdn.net/weixin_44170512/article/details/103990592(主要参考)

官方给出的,环境搭配:

https://tensorflow.google.cn/install/source_windows


先说下我的环境:

显卡:gtx 950M

python:3.6(ANACONDA 默认python3.7.4,我新建了python的3.6环境)

CUDA: 10.1

cuDNN :v7.6.5

目标:tensorflow_gpu 2.1.0


步骤:

1.安装Anaconda

Anaconda默认是pythonn3.7.4,tensorflow2.1版本可以使用python3.7.4

2.确定tensorflow是cpu版还是gpu版

cpu版和gpu版只是运算速度上有区别,如果显卡支持的话,可以选择gpu版(我gtx950M的显卡,选的gpu版)

3.安装CUDA,cuDNN,并配置环境变量

https://blog.csdn.net/weixin_44170512/article/details/103990592(这位大佬讲得很细!)


问题1:ImportError: DLL load failed: 找不到指定的模块。

上述都安装成功,也都测试正常。

但是,在我运行import tensorflow的时候,就会报错:ImportError: DLL load failed: 找不到指定的模块。

解决方法:

Tensorflow 2.1.0 运行需要安装 VS 2019 的环境,因为,单独安装了 VS 2015 或 VS 2017 的安装环境,均提示导入 Dll 错误,所以,需要下载同时包含,VS 2015、VS2017以及 VS2019 运行环境的安装包

下载地址:https://support.microsoft.com/zh-cn/help/2977003/the-latest-supported-visual-c-downloads

参考:https://blog.csdn.net/lingtianyulong/article/details/104207960,感谢大佬,这问题终于解决了。


问题2:CUDA_ERROR_UNKNOWN: unknown error

安装好tensorflow后,当然要测试下了,运行下面的代码

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

却报错:Attempting to fetch value instead of handling error Internal: failed to get device attribute 13 for device 0: CUDA_ERROR_UNKNOWN: unknown error

解决方法:更新显卡驱动就可以了


问题3:module 'tensorflow' has no attribute 'Session'

这其实不算问题,tensorflow的版本更新导致的

tensorflow是2.0以上版本(我用的是2.1)

sess = tf.Session()

已经改为:

sess = tf.compat.v1.Session()

tensorflow2.1的测试方法:

import tensorflow as tf

h = tf.constant('Hello, this is TensorFlow')
print(tf.print(h))

经过一段时间,会得到输出:

Hello, this is TensorFlow
None

 

你可能感兴趣的:(TensorFlow安装)