装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才可用
这一步不一定