官网安装说明
首先该链接非常重要,即使以后tensorflow更新了,我的博客不再适用了该链接依然可以助你安装成功

1.首先检查你的gpu

tensorflow只支持nvidia的显卡,AMD的同学很不幸下面没你事了
根据官网建议你的显卡的算力至少应该在3.5以上,
你可以在nvidia官网查看你的 gpu 的 compute capability

2.安装gpu driver

https://www.nvidia.com/download/index.aspx?lang=en-us
填写你的gpu信息后点search,如果没有搜索结果 do this:
最详细 tensorflow2.0 gpu 版安装全程指导
然后安装:
最详细 tensorflow2.0 gpu 版安装全程指导
最详细 tensorflow2.0 gpu 版安装全程指导
安装完成:
最详细 tensorflow2.0 gpu 版安装全程指导

3.安装CUDA

注意参考官网安装说明安装合适的版本
CUDA下载地址

最详细 tensorflow2.0 gpu 版安装全程指导

4.安装CUDNN

安装与CUDA相匹配的CUDNN版本:
cudnn下载地址
注意点 I agree后才会出现各个版本,并且cudnn需要登陆才能下载。
最详细 tensorflow2.0 gpu 版安装全程指导

5.配置CUDA与CUDNN的环境变量

环境配置参见官网安装说明

最详细 tensorflow2.0 gpu 版安装全程指导

6.安装TensorRT(可选)

注意不要下最新版!!!
注意不要下最新版!!!
注意不要下最新版!!!
谷歌跟不上!
TensorRT5.0下载链接

7.安装 Miniconda 而非Anconda(它比Anconda小很多,我比较喜欢)

Miniconda下载链接
8.miniconda安装tensorflow-gpu=2.0
直接安装就行

9.测试

最详细 tensorflow2.0 gpu 版安装全程指导
注意上面图片的最后一句花了大概好几分钟,后面的就快了,第一次运行可能需要加载一些东西。
用jupyter测试

最详细 tensorflow2.0 gpu 版安装全程指导
官方案例演示

相关文章:

  • 2021-05-14
  • 2021-12-08
  • 2021-08-04
  • 2021-08-06
  • 2022-01-17
  • 2021-09-15
  • 2021-06-06
猜你喜欢
  • 2021-06-11
  • 2022-12-23
  • 2021-10-19
  • 2022-01-05
  • 2022-01-14
  • 2021-07-11
  • 2021-10-23
相关资源
相似解决方案