安装CUDA+CUdnn+tensorflow-gpu过程及遇到的问题记录

以下为安装Tensorflow过程记录

一、安装CUDA视频教程

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文件

whl文件下载链接

第一步:
找到自己需要的的tensorflow-gpu版本
安装CUDA+CUdnn+tensorflow-gpu过程及遇到的问题记录_第1张图片

第二步:
找到你的python版本所对应的whl文件
安装CUDA+CUdnn+tensorflow-gpu过程及遇到的问题记录_第2张图片

第三步:
执行命令:

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成功!!!

你可能感兴趣的:(tensorflow,python,深度学习)