本文使用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 即表示安装成功。