装tensorflow cpu版本还是很简单的,装gpu版本则要一步步来,而且要仔细看报错。
电脑配置:
- win10 64位
- python3.6.7 64位
- Anaconda 2019.03版
- 华硕gtx1060 6g
主要步骤:
- 根据显卡安装cuda
- 根据cuda安装cudnn
- 安装tensorflow-gpu
主要下载的网站:
- cuda:https://developer.nvidia.com/cuda-toolkit-archive
- cudnn:https://developer.nvidia.com/rdp/cudnn-download
before:
在安装之前一定要先看看自己gpu的算力 https://developer.nvidia.com/cuda-gpus
可以看到我的显卡6.1的算力,算力大于3.0才可用
![]()
这一步不一定需要,有vs2015版本以上的应该可以

本文详细介绍了在Windows 10上安装和配置TensorFlow-GPU 1.11.0的步骤,包括安装CUDA、CUDNN以及解决各种安装过程中可能遇到的问题。内容涵盖根据显卡型号选择CUDA版本,CUDA和CUDNN的下载安装,以及如何避免和解决TensorFlow加载失败的错误。
最低0.47元/天 解锁文章
1197

被折叠的 条评论
为什么被折叠?



