以下为安装Tensorflow过程记录
一、安装CUDA视频教程
根据自己的需求下载相对应的版本!
步骤都是一样的!
二、安装CUdnn视频教程
CUdnn视频教程
下载自己需要的版本!(同上)
三、安装tensorflow
(1)安装cpu版本(此版本不用配置CUDA)
pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple
其中https://pypi.tuna.tsinghua.edu.cn/simple为清华镜像源(加快下载速度)
(2)安装GPU版本
方式一:pip指定特定版本直接下载
pip install tensorflow-gpu==2.4.1
可加清华镜像源加快速度: -i https://pypi.tuna.tsinghua.edu.cn/simple
方式二:下载对应版本的whl文件
第一步:
找到自己需要的的tensorflow-gpu版本
第二步:
找到你的python版本所对应的whl文件
第三步:
执行命令:
pip install tensorflow_gpu-2.9.2-cp37-cp37m-win_amd64.whl
安装成功后验证:
python
import tensorflow as tf
tf.constant(1.) + tf.constant(1.)
运行后:
四、遇到的问题
安装tensorflow-gpu时遇到的问题
(1)Command “python setup.py egg_info” failed …问题
解决方法:
pip install --upgrade setuptools
一开始没有用,查了很多资料,在运行pip install tensorflow_gpu-2.9.2-cp37-cp37m-win_amd64.whl
后,执行pip install --upgrade setuptools
,然后还有其他错误,但是都是可以在csdn一搜就能解决的。
(2)比如:
pip install 时 WARNING: No metadata found in e:\anaconda\lib\site-packages 问题解决
然后把报错的文件删除:
然后重新安装 pip install 安装即可。
注:最后成功解决,安装tensorflow成功!!!