20系列和30系列显卡下的tensorflow-gpu环境配置,避坑指南

此方法适合于20系列显卡环境的配置,较传统方法简便

参考博客
https://blog.csdn.net/m0_49090516/article/details/113576003?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522163711104416780255215587%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=163711104416780255215587&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~top_positive~default-1-113576003.pc_search_result_control_group&utm_term=tensorflow-gpu&spm=1018.2226.3001.4187

配置流程
安装anaconda3_2019.10版本
conda create -n tensorflow python=3.7
conda install cudatoolkit=10.1
conda install cudnn=7.6
pip install tensorflow-gpu==2.1.0


30系列的显卡使用上述方法安装上述版本会存在显卡启动特别慢的问题
30系列显卡使用上述方法安装更高版本cuda11.0和cudnn8.0时,cudnn难以安装
因此采用传统下载软件方法安装

参考博客
https://blog.csdn.net/typefree/article/details/117998734?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522163712486616780261961097%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=163712486616780261961097&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduend~default-1-117998734.pc_search_result_control_group&utm_term=rtx3050%E9%85%8D%E7%BD%AEtensorflow&spm=1018.2226.3001.4187

配置流程
安装anaconda3_2019.10
在英伟达的控制面板的系统信息,组件里面,查看自己的显卡适合的cuda版本
安装cuda11.4,安装时不要勾选vs的安装
安装cudnn8.2,如果下载不了,可以将下载链接放在迅雷里面下载,不需要注册https://developer.nvidia.com/rdp/cudnn-archive
将cudnn解压后的文件复制在cuda的安装路径里面,例如C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2
在电脑的系统变量里面添加环境变量
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4
打开anaconda prompt 创建环境变量
安装相对应的tensorflow-gpu版本   https://tensorflow.google.cn/install/source#gpu
 

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
您好!对于 TensorFlow-GPU 的验证,您可以按照以下步骤进行操作: 1. 首先,确保您的计算机已经安装了适当的 NVIDIA 显卡驱动程序,并且支持 CUDA 和 cuDNN。您可以在 NVIDIA 的官方网站上找到相应的驱动程序和支持的 CUDA 版本。 2. 接下来,安装适用于 TensorFlow-GPUCUDA 工具包。您可以从 NVIDIA 的开发者网站上下载并安装与您的显卡驱动程序兼容的 CUDA 版本。 3. 安装 cuDNN 库,这是用于加速深度神经网络训练的 NVIDIA 库。您需要从 NVIDIA 的开发者网站上下载并安装与您安装的 CUDA 版本兼容的 cuDNN 版本。 4. 创建一个虚拟环境(可选),并在其中安装 TensorFlow-GPU。您可以使用 conda、virtualenv 或者其他工具来创建和管理虚拟环境。 5. 在虚拟环境中使用 pip 命令安装 TensorFlow-GPU。例如,使用以下命令可以安装 TensorFlow-GPU 2.0 版本: ``` pip install tensorflow-gpu==2.0 ``` 6. 安装完毕后,您可以通过导入 TensorFlow 库并执行一些简单的代码来验证 GPU 是否正常工作。例如,您可以尝试运行以下代码: ```python import tensorflow as tf # 检查 GPU 是否可用 print(tf.test.is_gpu_available()) # 打印 GPU 设备名称 print(tf.config.list_physical_devices('GPU')) ``` 如果输出结果显示 GPU 可用,并列出了您的 GPU 设备信息,则说明 TensorFlow-GPU 已成功安装并与 GPU 正常配合工作。 请注意,确保您的硬件和软件配置符合 TensorFlow-GPU 的要求,并且按照官方文档提供的指南进行操作。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值