tensorflow-gpu 的安装(亲测有效)

如果你想要GPU 进行训练的话,那你必须要有三个东西:
1.tensorflow-gpu
2.cuda
3.cudnn
注意:tensorflow-gpu,cuda和cudnn这两个版本号有着相互依赖的关系
CUDA8.0 不支持tensorflow 1.5及以上,所以建议安装CUDA9.0
安装的话就是在官网下载安装
下载过之后:
在这里插入图片描述
需要安装这两个 ,然后将这两个分别解压安装,解压安装之后,重点来了:
!!!两点必须要做!!!

  1. 必须要将cudnn里面的bin,include,lib三个文件里面的具体的文件分别放入到安装之后的cuda里面,记住要一一对应,cudnn bin里面的文件要对应放入cuda里面的bin文件中等

    cudnn里面的三个文件
  2. 环境变量的配置:
    A.
    上面的是在安装好之后就有的,

    在这里插入图片描述
    这个上面是你要手动添加的 注意路径问题

通过之上的安装之后,就可以安装GPU了,亲测有效

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章