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

由于换电脑了,原电脑显卡太垃圾,跑不了GPU版本就一直没装,其实安装过程蛮简单的,只要按步骤来即可。

目录

一、安装Anaconda

二、安装cuda和cudnn

三、安装Tensorflow GPU版本

四、测试是否安装成功

五、在JupyterNotebook中配置tensorflow


一、安装Anaconda

Anaconda安装地址:官方下载地址

这一步傻瓜式操作即可,但注意需要配置Anaconda环境

激活一个虚拟环境

conda create -n TF_2C python=3.6
# 进入虚拟环境
conda activate TF_2G

二、安装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则安装成功

五、在JupyterNotebook中配置tensorflow

# 激活TF环境
conda activate TF_2G
# 安装jupyter和ipython
conda install ipython
conda install jupyternotebook
# 测试内核
jupyter notebook

# 在对应窗口输入
import tensorflow as tf

猜你喜欢

转载自blog.csdn.net/weixin_40539952/article/details/106883060
今日推荐