conda环境下tensorflow-gpu的安装

本文详细介绍了如何在Windows上检查并安装适合的NVIDIA驱动程序,选择并安装CUDA 11.2,包括CUDA工具包的下载与定制安装,cudnn的下载与整合,以及最后TensorFlow-GPU的安装和验证。遇到的兼容性和警告问题也做了说明。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

查看安装的驱动程序

打开nvidia控制面板->系统信息,查看GPU的型号,我的型号为GeForce GTX 1660 Ti,驱动程序版本为462.42
在这里插入图片描述
打开组件查看安装的GPU驱动版本,显示我的显卡支持版本为NVIDIA CUDA 11.2.162版本
在这里插入图片描述https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html中查看windows下CUDA版本于安装驱动程序的版本对应关系
在这里插入图片描述
我的驱动程序版本为462.42,可以支持CUDA11.3.0版本以下的版本,我选择的CUDA版本为CUDA 11.2.2。
https://tensorflow.google.cn/install/source_windows?hl=en#gpu查看tensorflow-gpu安装的相应版本关系
在这里插入图片描述
这里我选择安装tensorflow_gpu-2.5.0版本,各个版本对应为
在这里插入图片描述

安装CUDA

安装CUDA工具包

下载CUDA工具包,以前版本的网页查找以前的版本,点击CUDA Toolkit 11.2.2,如下图
在这里插入图片描述
跳到页面选择下载windows系统的CUDA,选择如下图下载安装工具
在这里插入图片描述

下载完成,双击打开进行安装
在这里插入图片描述
选择下载器的安装路径,默认即可
在这里插入图片描述
出现安装界面
在这里插入图片描述
选择自定义
在这里插入图片描述
选择下载的组件,默认全选,我已经安装了驱动组件,这么不选择下载驱动组件,其他组件也取消
在这里插入图片描述
CUDA组件里选择下面几项即可,我还没有下载vs,不勾选Visual Studion Integration
在这里插入图片描述

选择安装路径,这里默认路径即可,点击下一步进行下载
在这里插入图片描述
安装完成
在这里插入图片描述

查看安装

在我的安装路径已经多出了一个CUDA文件夹,
在这里插入图片描述

点击计算机右键属性->高级系统设置->环境变量->系统变量如下图
在这里插入图片描述

此时CUDA环境已经自动配置成功,win+R 输入cmd打开命令行
输入nvcc --version查看CUDA,显示如下则安装成功
在这里插入图片描述

安装cudnn

去官网下载cudnn,需要登陆。
在这里插入图片描述

选择相应的版本,我下载的是8.1.1版本
在这里插入图片描述
选择cuDNN Library for Windows(x86)进行下载,下载的文件是一个压缩包,需要解压,解压后的文件夹名为cuda,里面文件如下
在这里插入图片描述
将cuDNN里面的下面三个文件夹下的文件分别复制到之前安装的CUDA的相应同名文件下。
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
cudnn安装成功

安装tensorflow-gpu

win+R 输入cmd打开控制台,输入如下命令安装tensorflow-gpu-2.5.0(需要此前已经配置成功anaconda环境)
在这里插入图片描述
如下图所示下载成功
在这里插入图片描述

确实tensorflow安装成功

conda list 发现存在多出如下几个包
在这里插入图片描述直接进入python环境,输入import tensorflow as tf导包,显示如下则安装成功在这里插入图片描述

补充

若是在导包中出现以下警告
在这里插入图片描述
则是cuda的一些库没有找到,cuda的库版本是向下兼容,如果是版本高了,可以在cuda安装目录中的bin目录里面进行重命名为需要找到的链接库即可

在这里插入图片描述

### 如何通过 Conda 安装 TensorFlow-GPU 为了确保能够正确安装 `tensorflow-gpu` 并使其支持 GPU 加速,以下是详细的说明: #### 创建并激活虚拟环境 创建一个新的 Python 虚拟环境可以有效隔离依赖项冲突。假设目标是使用 Python 3.9 和 TensorFlow 2.10 版本,则可以通过以下命令完成环境的创建和激活: ```bash conda create -n TF2.10 python=3.9 conda activate TF2.10 ``` #### 安装 TensorFlow-GPU 在激活环境中,可以直接通过 `conda` 命令来安装 `tensorflow-gpu`。需要注意的是,某些情况下可能需要指定特定版本号以及镜像源以加速下载过程。例如: ```bash conda install tensorflow-gpu==2.10.0 -c anaconda ``` 如果遇到网络问题或者速度较慢的情况,可以选择清华 TUNA 镜像作为替代方案: ```bash conda install tensorflow-gpu==2.10.0 -i https://pypi.tuna.tsinghua.edu.cn/simple ``` 对于更复杂的场景下(比如已经存在其他包冲突),还可以尝试先卸载再重新安装的方式解决潜在问题: ```bash conda uninstall tensorflow-gpu==2.10.0 conda install tensorflow-gpu==2.10.0 -i https://pypi.tuna.tsinghua.edu.cn/simple ``` 另外一种方法则是利用阿里云 PyPI 镜像源设置全局默认地址后再执行常规安装操作[^3]: ```bash pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/ pip install tensorflow==2.10 ``` 以上步骤均基于前提条件——计算机已预装兼容版本的 NVIDIA CUDA 工具链及相关驱动程序。 ### 注意事项 - 确认硬件设备满足最低需求规格。 - 检查当前系统的显卡驱动是否适配所选CUDA版本。 - 如果仍然存在问题,请参照官方文档进一步排查原因[^2]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值