| |
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
-> Python知识库 -> 【彻底解决】【多版本共存】安装 Tensorflow Pytorch 环境 遇到的 nvidia CUDA Toolkit cuDNN 问题 -> 正文阅读 |
|
[Python知识库]【彻底解决】【多版本共存】安装 Tensorflow Pytorch 环境 遇到的 nvidia CUDA Toolkit cuDNN 问题 |
整体上有两种方案:1. 手动安装 nvidia 提供的 CUDA、cuDNN 组件包(多见于tensorflow的安装) 2. 安装 cuda 提供的 cudatoolkit 精简运行包(多见于pytorch的安装) 方案1:手动安装 nvidia 提供的CUDA组件包Tensorflow 版本及依赖tensorflow 与 CUDA、cuDNN 对照表: win版本对照:https://www.tensorflow.org/install/source_windows linux版本对照:https://www.tensorflow.org/install/source#gpu_support_3 注意: 当前最新版本如果未列出,则继续使用列表中最高版本的cuda和cudnn版本 CUDA版本不要追求最新,一般满足版本要求的最低版本,可以一次性安装成功,带有小版本的升级版本反而会出现安装错误 一般表中给出的是 X.X,默认指的是 X.X.0 ,选择 CUDA 版本都一致的,即使小版本号不一致,也会导致安装失败 NVIDIA CUDA Toolkit 安装CUDA Toolkit 也即 CUDA nvidia 版本确认: 在 CMD/ shell 中输入:nvidia-smi 其中,看到的 cuda 版本只是驱动程序的编译版本,实际 tf 不一定会调用该?cuda 版本 这里主要关注的是驱动程序的版本号 GPU驱动程序与 CUDA 版本之间的匹配关系:Release Notes :: CUDA Toolkit Documentation CUPTI默认位于CUDA中,需要环境变量指定 CUDA安装 CUDA默认安装
安装时若遇到 “You already have a newer version of the NVIDIA Frameview SDK installed” 先把电脑已经存在的FrameView SDK 卸载掉,把C:\Program Files\NVIDIA Corporation\FrameViewSDK文件夹删掉 NVIDIA cuDNN 安装cuDNN安装 方式一:将cuDNN解压后直接复制到CUDA路径中覆盖 方式二:cuDNN 安装至其他位置
环境变量配置powershell快捷配置
手动环境变量配置 系统中多个运行环境并存,通过修改全局环境变量CUDA_PATH、CUDNN_PATH实现运行环境切换 分别定义不同版本的CUDA环境变量,
统一定义一个全局变量,用于环境变量的切换,实现切换不同运行环境的功能
添加到PATH中
cuDNN 采用方式二安装时,需要配置环境变量 分别定义不同版本的cuDNN环境变量
统一定义一个全局变量,用于环境变量的切换,实现切换不同运行环境的功能
添加到PATH中
? 安装结果验证查看 nvidia 版本 这里显示的是 CUDA Driver 版本 nvidia-smi 注意:显卡驱动自带一个 CUDA Driver ,其版本 >= CUDA RunTime 版本 ,否则就会报错 查看 cuda 核心版本: 这里显示的是 CUDA RunTime 版本 nvcc -V 验证 cudnn 运行cuda 安装路径下 extras/demo_suite/下 deviceQuery.exe 和 bandwidthTest.exe 两个程序 两个都显示为 PASS 表示验证通过 ? tensorflow 安装推荐使用pip安装,conda安装会导致奇奇怪怪的问题
验证
查看GPU是否可用:
报错: import tensorflow as tf 时报错 ImportError: DLL load failed: 找不到指定的模块。 替换CUDA版本解决 如果 python 中无法使用GPU训练模型,提示 找不到 cudnn_x64_8.lib,则表示cuda版本选错了, 需要重新安装 CUDA 并且 pip 卸载 tensorflow-gpu keras 并重新安装 安装 keras(可选,一般tf自带的keras就够用了)
方案二:安装 cuda 提供的 cudatoolkit 精简运行包使用 conda 安装 cudatoolkit两者区别: 1.官方版本,包含完整的编译测试工具和依赖,以及编译好的动态链接库 2. conda版本,仅包含编译好的动态链接库。大多数直接调用的情况下(tf,pytorch)可运行,少部分需要编译的情况时需要完整的CUDA toolkit包,就会报错。 安装
安装 Pytorch
参考文章: 显卡,显卡驱动,nvcc, cuda driver,cudatoolkit,cudnn到底是什么? - marsggbo - 博客园 conda安装的cudatoolkit与Nvidia官方提供的cudatoolkit的区别_马圈圈马的博客-CSDN博客_conda cudatoolkit与cuda的区别 |
|
|
上一篇文章 下一篇文章 查看所有文章 |
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 | -2024/11/15 10:47:18- |
|
网站联系: qq:121756557 email:121756557@qq.com IT数码 |