Python中使用tensorflow-gpu的配置方法

部署运行你感兴趣的模型镜像

Python中使用tensorflow-gpu的配置方法:

一、下载安装python 3.8:
登录tensorflow官网,查看支持GPU的python版本,然后下载对应的python版本并安装(https://www.python.org/downloads/windows/),然后再设置python的安装路径到系统变量的path中,注意两个路径:C:\Users\Administrator\AppData\Local\Programs\Python\Python38和C:\Users\Administrator\AppData\Local\Programs\Python\Python38\Scripts,打开cmd窗口,输入:python --version,如果版本正确就说明安装正确。我安装的python版本是3.8,是可以的
二、使用tensorflow的系统要求:
1、支持tensorflow的Python版本: 3.6–3.9
若要支持 Python 3.9,需要使用 TensorFlow 2.5 或更高版本。
若要支持 Python 3.8,需要使用 TensorFlow 2.2 或更高版本。我安装的是2.4.4是可以的
pip 19.0 或更高版本(需要 manylinux2010 支持)
2、Ubuntu 16.04 或更高版本(64 位)
3、macOS 10.12.6 (Sierra) 或更高版本(64 位)(不支持 GPU)
macOS 要求使用 pip 20.3 或更高版本
4、Windows 7 或更高版本(64 位)
适用于 Visual Studio 2015、2017 和 2019 的 Microsoft Visual C++ 可再发行软件包
5、GPU 支持需要使用支持 CUDA® 的卡(适用于 Ubuntu 和 Windows)
三、安装显卡最新驱动程序:NVIDIA Geforce GTX 1050Ti,如果显卡正常则不需要安装。当前版本:30.0.14.7212, NVIDIA驱动程序:确保您的NVIDIA图形卡驱动程序是最新的,这样可以确保与CUDA和cuDNN的良好兼容性。
四、CUDA Toolkit 11.4:这是NVIDIA提供的用于GPU计算的开发工具包。TensorFlow使用CUDA来进行GPU计算。您需要安装与您希望安装的TensorFlow版本兼容的CUDA版本。您可以在NVIDIA官网下载CUDA。
五、cuDNN 11.4:cuDNN是专为深度神经网络设计的GPU加速库。它作为CUDA的一部分安装。同样地,您需要与CUDA和TensorFlow版本兼容的cuDNN版本。同样可以在NVIDIA官网获取cuDNN。
安装完这些依赖后,您可以尝试安装tensorflow-gpu。如果pip在安装过程中遇到任何依赖问题,它会通知您。
六、Python 3.8对应的tensorflow版本2.4.4(我的笔记本电脑用这个版本是可以的):可以登录:https://www.tensorflow.org/进行查询
官网Python3.8对应的tensorflow版本下载位置:https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.6.0-cp38-cp38-win_amd64.whl ,然后再使用pip进行安装下载的包即可。
七、在pycharm中使用anaconda3中的python.exe解释器的方法:在设置-你的项目-解释器-增加解释器中选择conda,具体可以自己百度。
八、如果你在使用 Anaconda 时遇到了 Python 版本不匹配的情况,比如 Anaconda 默认安装的是 Python 3.11,但是你的项目要求使用 Python 3.8,你可以采取以下几种方法来解决这个问题:
1、创建一个新的 Conda 环境:
创建一个新的 Conda 环境,并指定 Python 3.8 的版本。这不会影响你的全局 Python 安装,并且允许你为特定的项目使用特定版本的 Python。
在 Anaconda Prompt 或者终端中运行以下命令来创建新的环境:conda create --name myenv python=3.8
2、激活新环境:conda activate myenv,注意使用pycharm设置虚拟环境时,不用激活,pycharm会自动激活。
九、创建虚拟环境以隔离不同的项目的环境:有两种方法:1、使用python自带的visualenv来创建虚拟环境;2、使用anaconda来创建虚拟环境。
创建方法:
第一种使用pycharm的sets设置:项目-python Interpreter-Add Interpreter-Virtualenv Environment-new
第二种使用pycharm的sets设置:项目-python Interpreter-Add Interpreter-Conda Environment-new
注意:使用第二种方式创建虚拟环境时不需要单独再去激活这个虚拟环境。pycharm会自动来处理。
十、在运行窗口中输入“NVIDIA Control Panel”,然后点击”OK”或回车键,打开NVIDIA控制面板。
在NVIDIA控制面板中,找到“帮助”菜单,并点击它。
在帮助菜单中,选择“系统信息”选项。这将打开一个新窗口,显示有关您的NVIDIA显卡的详细信息。
在系统信息窗口中,找到“组件”部分,并展开它。在这里,您将看到有关CUDA驱动程序的版本信息。记下CUDA驱动程序版本号。
我的笔记本电脑的显卡对应的cuda版本号:11.4.136,那么对应的CUDA ToolKit的版本是:11.4或小于这个版本号,试过11.4不行,又下载了11.0是可以的
十一、安装好CUDA Toolkit 11.0后设置系统变量:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\lib\x64
十二、验证CUDA Toolkit 11.0是否安装成功:nvcc --version 如果成功返回版本号就说明安装成功。
十三、CUDA Toolkit 11.0对应的cuDNN的版本是:cuDNN V8.2.4
下载地址:https://developer.nvidia.com/rdp/cudnn-archive
文件名:cudnn-11.4-windows-x64-v8.2.4.15.zip
十四、我的笔记本的tensorflow-gpu配置如下:
python:3.8
tensorflow-gpu:2.4.4
cuda Toolkit:11.0
cuDNN:11.4
opencv-python:4.10.0
numpy:1.21.0
经测试完全可以调用GPU

秋风写于淄博,业务联系与技术交流:Q375172665

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

### 安装 TensorFlow GPU 版本于 Python 3.7.6 环境 对于希望在 Python 3.7.6 的环境中安装 TensorFlow GPU 版本来利用图形处理单元加速机器学习模型训练的开发者来说,有几个重要的注意事项和步骤。 #### 验证硬件兼容性和驱动程序更新 确保计算机配备有支持 CUDA 的 NVIDIA 显卡,并且已安装最新的显卡驱动程序。Windows 下建议至少安装 CUDA 10.1 或更高版本以获得更好的兼容性和支持[^4]。 #### 安装必要的依赖库 为了使 TensorFlow 能够有效运行并充分利用 GPU 加速功能,还需要预先配置一些额外组件: - **CMake**: 可通过 `pip` 工具来完成安装操作,即执行命令 `pip install cmake` 来获取最新稳定版 CMake[^1]。 - **CUDA Toolkit 和 cuDNN SDK**: 这些工具包提供了实现高效计算所需的底层 API 接口和服务。按照官方文档指导选择合适的版本组合进行下载与设置。 #### 创建虚拟环境(可选) 考虑到不同项目可能有不同的依赖需求,推荐使用 Anaconda 创建独立的 Python 环境来进行实验或开发工作。这有助于隔离各个项目的软件栈,减少冲突风险。 ```bash conda create -n tf_gpu_env python=3.7.6 conda activate tf_gpu_env ``` #### 使用 Conda 渠道简化安装过程 借助 Anaconda 提供的强大包管理系统可以直接从其默认频道或其他可信源快速部署所需资源。下面给出了一种简便的方式用于一次性解决大部分依赖关系: ```bash conda install tensorflow-gpu cudatoolkit=10.1 cudnn ``` 此方法不仅能够自动拉取匹配当前系统的 TensorFlow GPU 库及其关联项,还能显著降低手动调整参数带来的不确定性。 #### 测试安装成果 最后,在确认所有前置条件均已满足之后,可以通过简单的测试脚本来验证 TensorFlow 是否成功识别到了可用的 GPU 设备: ```python import tensorflow as tf print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU'))) ``` 如果一切正常,则会显示检测到的一个或多个人工智能专用处理器;反之则需仔细排查上述各环节是否存在遗漏之处。
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值