tensorflow-gpu 下载和安装

川长思鸟来 2022-04-22 15:04 342阅读 0赞

如果你想要GPU 进行训练的话，那你必须要有三个东西：
1.tensorflow-gpu
超时的话pip install tensorflow-gpu==1.13.1 -i https://pypi.douban.com/simple
2.cuda
CUDA其他版本下载地址是https://developer.nvidia.com/cuda-toolkit-archive
3.cudnn
下载地址是：https://developer.nvidia.com/rdp/cudnn-archive，在下载CUDNN7.0之前，会提示你注册一个账号，输入邮箱等相关注册信息注册一个账号
注意：tensorflow-gpu，cuda和cudnn这两个版本号有着相互依赖的关系
CUDA8.0 不支持tensorflow 1.5及以上，所以建议安装CUDA9.0
安装的话就是在官网下载安装
下载过之后：
在这里插入图片描述
需要安装这两个 ，然后将这两个分别解压安装，解压安装之后，重点来了：
！！！两点必须要做！！！：

  1. 必须要将cudnn里面的bin，include，lib三个文件里面的具体的文件分别放入到安装之后的cuda里面，记住要一一对应，cudnn bin里面的文件要对应放入cuda里面的bin文件中等
    20190114100823498.png
    cudnn里面的三个文件
  2. 环境变量的配置：
    A.20190114100423903.png
    上面的是在安装好之后就有的，
    20190114100549583.png
    在这里插入图片描述
    这个上面是你要手动添加的 注意路径问题

通过之上的安装之后，就可以安装GPU了，亲测有效在这里插入图片描述

发表评论

表情：
评论列表 （有 0 条评论，342人围观）

还没有评论，来说两句吧...

相关阅读