windows下升级TensorFlow到2.1.0
内容导读
互联网集市收集整理的这篇技术教程文章主要介绍了windows下升级TensorFlow到2.1.0,小编现在分享给大家,供广大互联网技能从业者学习和参考。文章包含1373字,纯文字阅读大概需要2分钟。
内容图文
1、pip install tensorflow-gpu==2.1.0 --user(pip install tensorflow-gpu==2.1.0 -i https://pypi.tuna.tsinghua.edu.cn/simple --user)
2、安装VC_redist.x64.exe:
安装
2、import tensorflow出错:
Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found
解决办法:https://cn.dll-files.com/cudart64_101.dll.html
下载cudart64_101.zip
解压后把dll文件拷贝到:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
3、import成功:
>>> import tensorflow
2020-03-23 15:37:43.258666: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_101.dll
>>> tensorflow.__version__
'2.1.0'
4、升级cuda到10.1版本:https://developer.nvidia.com/cuda-downloads 下载:cuda_10.1.105_418.96_win10,自定义安装。
安装成功后,验证:
import tensorflow as tf
#验证cuda
a = tf.test.is_built_with_cuda()
print(a)
physical_devices = tf.config.experimental.list_physical_devices('GPU')
print(physical_devices)
5、升级cudnn到7.6.5:
安装cudnn7.6.5 for cuda10.1 for windows,下载:cudnn-10.1-windows10-x64-v7.6.5.32.zip,https://developer.nvidia.com/rdp/cudnn-download
把bin、include、lib里面的文件分别复制到cuda安装目录对应子目录。
内容总结
以上是互联网集市为您收集整理的windows下升级TensorFlow到2.1.0全部内容,希望文章能够帮你解决windows下升级TensorFlow到2.1.0所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错,欢迎将互联网集市网站推荐给程序员好友。
内容备注
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 gblab@vip.qq.com 举报,一经查实,本站将立刻删除。
内容手机端
扫描二维码推送至手机访问。