欢迎光临散文网 会员登陆 & 注册

TensorFlow 安装环境搭建

2023-04-09 10:57 作者:雨的印记_  | 我要投稿


目录

  1. 安装Anaconda

  2. 安装CUDA

  3. 安装cuDNN

  4. 配置环境变量

  5. 安装TensorFlow

1.  安装Anaconda

  1. Anaconda 下载:https://www.anaconda.com/products/distribution/start-coding-immediately

  2. 安装到D盘,节省空间


选择安装位置

3.继续勾选下面的选项


勾选图中选项


4. 安装需要一些时间


安装中


5. 安装成功后,按住键盘上的win+R,打开运行,输入cmd回车。打开命令行窗口,输入:conda list,检查是否成功安装


安装成功
检查是否安装成功,输入 conda list

2. 安装cuda

- 检查是否是`NVIDIA`显卡

  1. 按住键盘上的ctrl+Shift+Esc,打开任务管理器,选择性能一栏,查看GPU。


查看GPU

2. 我的显示有一块NVIDIA显卡,可以安装CUDA

-下载CUDA

  1. 下载CUDA,依照我的电脑版本,选择了win10,本地


下载CUDA

2. 安装CUDA,我选择临时解压在D盘

选择解压位置


3. 安装选项:

同意并继续
选择自定义
取消勾选VS,若安装了VS则勾选
是否安装新版本驱动

            4.选择安装位置

选择安装位置

    5.    安装完成后,打开安装文件夹D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\bin下的==nvcc.exe==,有该文件存在安装基本

检测安装是否成功

6. 打开D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\extras\CUPTI\lib64,确认是否存在==cupti64_2023.1.0.dll==动态库


检测安装是否成功


    3. 安装cuDNN

    1. 下载cuDNN,选择与CUDA相匹配的版本,我这里安装的CUDA v12.0,故选择cuDNN v8.8.1 (March 8th, 2023), for CUDA 12.x

找到合适版本
下载cuDNN

    2.  解压压缩包,并将解压后的文件夹改名为cudnn,并复制到D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1文件夹下。

解压文件
复制到该文件夹,并重命名

    3.确认D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\cudnn\bin文件夹下是否有cudnn64_8.dll,这关系到tensorflow_GPU版能否安装成功。

检查是否安装成功


    4.配置环境变量

    1、打开环境变量设置,如图所示。

打开环境变量

        2、找到path,确定两项是否存在

检查path下的两条地址是否存在

    3、添加其余两项,并将这四项置顶,否则会影响搜索,切四项缺一不可

添加红框中的两条地址并利用上移来置顶

    4、测试CUDA,打开命令行窗口,输入nvcc -V,显示版本号,则说明安装成功

检测是否成功


    5. 安装TensorFlow

    1. 打开Anaconda Prompt窗口


打开Anaconda 命令行窗口

2. 安装GPU版本的tensorflow,在命令行窗口输入conda install tensorflow
因为旧的tensorflow-GPU已经被删除了,目前只有tensorflow这个版本,也可以使用GPU计算,参考此连接https://pypi.org/project/tensorflow-gpu/

3.测试tensorflow是否安装成功

检测是否安装成功




TensorFlow 安装环境搭建的评论 (共 条)

分享到微博请遵守国家法律