tensorflow2.0 GPU版本安装

casdfxx

已于 2022-12-25 18:50:39 修改

阅读量1k

点赞数

分类专栏： python与人工智能文章标签： tensorflow python 深度学习

于 2022-10-29 01:35:43 首次发布

本文链接：https://blog.csdn.net/c_1969/article/details/127582024

版权

python与人工智能专栏收录该内容

6 篇文章 2 订阅

订阅专栏

使用NVIDIA显卡，算力要求大于3.5

step1、

通过miniconda管理安装环境，优点是自动安装依赖，方便快捷。

Miniconda — conda documentation

step2、

在搜索栏中搜索miniconda点击进入。

输入：

conda install tensorflow-gpu-2.6.0

安装的内容较多，耐心等待。使用加速器更快。

step3、

输入：

pip install pandas matplotlib notebook keras==2.6.0 -i https://pypi.douban.com/simple/

红色字体显示正常

正常网络环境下，安装所需时间不多。

step4、

以上安装完成后，GPU还是不能用，那该怎么办？

需要在miniconda（anaconda）中配置两个CUDA的组件：

conda install -c anaconda cudatoolkit
conda install -c anaconda cudnn
在下载第二个时，消耗时间过多，甚至到20%时网络原因（是服务器的原因不是我网络的原因）导致下载失败。所以找到一篇文章（https://blog.csdn.net/LIWEI940638093/article/details/113811563）有镜像：
        -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
这个镜像用着还行。