win10 安装 Tensorflow 2.60

本文使用Anaconda环境安装tensorflow 2.60

1.安装Anaconda

百度一下安装教程有很多~
https://blog.csdn.net/ITLearnHall/article/details/81708148
我安装的版本为 Anaconda3-2021.05-Windows-x86_64.exe

2.安装CUDA(如果电脑有NVIDIA显卡的话)

下载地址:https://developer.nvidia.com/cuda-toolkit-archive

注意下载的CUDA版本要与Tensorflow 2.60 对应,不然无法识别GPU!!!
版本对应如下:
在这里插入图片描述
根据上表tensorflow2.6 应该下载 11.2 版本
在这里插入图片描述
下载完成后,选择自定义安装,然后一直下一步即可~

3.安装cudnn

下载地址:https://developer.nvidia.com/rdp/cudnn-download (需要登陆)

必须选择与CUDA 11.2兼容的版本
在这里插入图片描述
下载完成后,解压到当前文件夹,并且重命名为 cudnn,然后将其拷贝到目录:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2
在这里插入图片描述

4. 添加环境变量

在系统环境变量PATH中添加两条新的环境变量,并上移到下图位置后保存即可。
在这里插入图片描述
打开 CMD 窗口,输入 nvcc -V 查看是否安装成功。
在这里插入图片描述

5.安装tensorflow

在CMD中执行以下命令进行安装(pip版本要最新的,本文使用的是Anacoda的python环境)

pip install tensorflow

新版tensorflow将cpu版本和GPU版本合并了
安装完成后显示:
在这里插入图片描述

6.测试

在CMD窗口中使用 ipython 进行测试,如下图所示
在这里插入图片描述
能正常导入TensorFlow 并且is_gpu_available()函数返回 True 即表示安装成功。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值