Tensorflow2.0安装全流程-2020最新

由于换电脑了,原电脑显卡太垃圾,跑不了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则安装成功

 

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