安装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__))
成功则不会报错!!!
文章出处登录后可见!
已经登录?立即刷新