安装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和cuda\cudnn版本的关系等等

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

如:

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

成功则不会报错!!!

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
xiaoxingxing的头像xiaoxingxing管理团队
上一篇 2022年5月22日
下一篇 2022年5月22日

相关推荐