一、安装tensorflow的cpu版本
pip install tensorflow
默认安装的是2.0版本(2019.10.08),安装完成,即可使用!
二、安装tensorflow-gpu版本
本人显卡GTX1060
python版本3.7.3
特别注意:需要去查看tensorflow、cuda和cudnn版本支持关系
显卡支持型号查询:https://developer.nvidia.com/cuda-gpus
tensorflow和cuda以及cudnn版本对应关系:https://blog.youkuaiyun.com/qq_27825451/article/details/89082978
1.下载CUDA并安装,选择自定义-只勾选CUDA即可
CUDA下载地址:https://developer.nvidia.com/cuda-toolkit-archive

2.下载CuDNN(必须与cuda版本对应),解压,按照下图操作

本文详细介绍了如何在Windows系统中分别安装TensorFlow的CPU和GPU版本。对于GPU版本,重点强调了检查显卡兼容性、选择合适的CUDA和CuDNN版本、安装过程及配置环境变量的步骤,最后通过测试确保安装成功。
最低0.47元/天 解锁文章
685





