由于换电脑了,原电脑显卡太垃圾,跑不了GPU版本就一直没装,其实安装过程蛮简单的,只要按步骤来即可。
一、安装Anaconda
Anaconda安装地址:官方下载地址
这一步傻瓜式操作即可,但注意需要配置Anaconda环境
二、安装cuda和cudnn
cudnn其实就是cuda的一个补丁
注意安装cuda10.0的版本,tensorflow还不支持11.0以上的cuda,cudnn选择cuda匹配的即可。
cuda下载地址:官方下载地址
cudnn下载地址:官方下载地址
- cuda安装傻瓜式操作,注意选择自定义安装,其中不要选择以下几项。
CUDA-》Visual Studio Integeration / Nsight Compute
- cuda安装完毕后解压cudnn
- 将所有目录复制到cuda目录中,完整目录如下
- 配置环境变量
三、安装Tensorflow GPU版本
pip install -U tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple # 注意报错就多装几次
四、测试是否安装成功
#以下是在python环境中
import tensorflow as tf
tf.test.is_gpu_available()
# 返回结果为True则安装成功