1、打开anaconda prompt 执行命令行 pip install tensorflow-gpu
2、查看CUDA版本和cuDNN版本。
一、显卡型号
桌面右键→NVIDIA控制面板→帮助→系统信息
二、CUDA Toolkit
- 查找显卡对应的CUDA版本
本文介绍了如何在Anaconda环境中通过pip安装Tensorflow-GPU，并详细阐述了如何确定并安装匹配的CUDA Toolkit和cuDNN版本。首先，查看NVIDIA控制面板获取显卡信息，然后按照显卡型号选择合适的CUDA版本进行下载安装。安装过程中需注意自动配置环境变量。接着，下载与CUDA版本相匹配的cuDNN，并将其路径添加到系统环境变量中。最后，对于初次使用TensorFlow-GPU的用户，文章提到了在创建Session时应配置GPU选项以优化内存使用。
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
