当前位置:首页 >> 新闻资讯 >> Python >> windows成功安装tensorflow的方法

windows成功安装tensorflow的方法

作者:本站 时间:2020-5-23 点击:

1、安装64位Python3.7 

笔者这里安装的是3.7.7版本,一定要选64位安装包,tensorflow在32位python下基本是无法安装成功的。

2、下载tensorflow的whl。

whl文件很大,镜像网络安装总是网络原因安装失败,下载后安装。

打开网址,https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple/tensorflow/ 
找到“tensorflow-2.2.0-cp37-cp37m-win_amd64.whl"下载下来

3、安装下载好的whl 

打开cmd
切换到whl文件的目录
输入命令,pip install -i https://pypi.tuna.tsinghua.edu.cn/simple "tensorflow-2.2.0-cp37-cp37m-win_amd64.whl"

4、如果系统未安装GPU

可在此网址下载

Nvidia CUDA系列 https://developer.nvidia.com/cuda-downloads

5、CUDA的dll名称修改

找到CUDA的安装目录(默认:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA)
直接搜索:cudart64
搜到的是cudart64_XXX.dll右键属性,把它改成cudart64_100.dll

6、代码测试

import tensorflow as tf
tf.compat.v1.disable_eager_execution() #保证sess.run()能够正常运行
hello = tf.constant('hello,tensorflow')
sess= tf.compat.v1.Session()#版本2.0的函数
print(sess.run(hello))


(以上内容由萍缘网络工作室收集整理自大量网站,转载请注明出处,如有侵权,通知后立即删除)