栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

安装tensorflow-gpu-1.14.0+cuda10.0+cudnn7.4.0+python3.7

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

安装tensorflow-gpu-1.14.0+cuda10.0+cudnn7.4.0+python3.7

安装tensorflow-gpu-1.14.0+cuda10.0+cudnn7.4.0

安装好acanada3,然后使用命令conda安装。

tensorflow(GPU、CPU):

gpu 版本和cpu版本不一样。CPU版本安装起来简单,没有cuda和cudnn的安装。但是GPU跑起来更快。

具体自己的电脑支持什么样的cuda,上网额外查。
其次tensorflow-gpu 和python 以及cuda、cudnn之间都有版本对应关系.这个根据自己的坏境进行安装和选择。

安装
conda install cuda=10.0

conda install cuda=7.4.0

pip install tensorflow-gpu==1.14.0

你以为这么简单??

不是的,首先,这些东西服务器在国外,国内有墙,连接不到,没法安装。这个时候就需要使用国内镜像网站,如清华的。

其次,镜像网站没有,比如我要安装的cudnn=7.4.0就没有,这个可以使用

conda search cudnn

来查看是否有你想要的的版本。如果没有,换镜像源;镜像源也没有,那么只能手动去这个官网下载exe文件。nvdia官网

好了,上面每个架构都讲清楚了。子问题网上找吧,依据关键词,一大堆,也不知道哪个有用,就只能怎么办呢?一个个试试,有时候试坏了,全部重头再来吧!!

关键词比如:

  • conda换镜像

  • cuda版本和GPU支持

  • gpu和cudacudnn版本的关系等等

本文给出总体思路,大家安装后,记得验证一下是否成功。

如:

import tensorflow as tf
print("TensorFlow version {}".format(tf.__version__))

成功则不会报错!!!

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/918575.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号