重装CUDA和cuDNN(目的是装TensorFlow和pytorch)【个人梳理总结】

本文是个人对重装CUDA和cuDNN以安装TensorFlow及PyTorch的步骤梳理。内容包括检查显卡驱动、下载CUDA和cuDNN、安装VScode,以及安装TensorFlow和PyTorch的方法,强调了阅读官方文档和理解软件需求的重要性。

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

写在前面

  以下是本人在学习过程中,在自己有限的阅读和学习后总结的重装/安装 CUDA、cuDNN、tensorflow和pytorch的框架式(流程式)经验。
  阅读完我的总结,你可以了解“到哪儿”,“找什么”,“什么顺序”以及“有哪些地方”等收获,同时也会留下很多自己在阅读官方文档和指引中的疑问帮助自己日后精进。
  我不倾向于分享给大家“无脑跟着做”或“保姆级”模式的教程,不利于个人对资料查找能力的成长,也不利于对软件的理解。
  CUDA、cuDNN以及VScode的安装与重装可靠性比较强,在一段时间内具有较大通用价值;末尾的TensorFlow和pytorch安装说实在的我也没搞明白,但是也分享了一些途径供大家当实在没有其他办法时尝试。

一、检查&下载驱动:

1、看自己的显卡驱动版本:

  使用组合键“Windowskey+R”启动“cmd”命令行,键入“nvidia-smi”回车,查看。
查看显卡信息

2、在NVIDA官网下载合适的显卡驱动并安装:(无需安装则跳过)

  如果需要重新安装驱动,记得准备好你的电脑硬件信息。
NVIDIA驱动程序下载
NVIDIA驱动程序下载

二、查找&下载CUDA和cuDNN:

1、检查显卡算力:

  执行这个操作的目的我理解有两个:

  • 你的显卡是否有GPU支持
  • 你的显卡算力是否值得装CUDA和cuDNN

说实在的,如果显卡算力不是很突出,你又是在学习阶段,那就没必要用GPU。也就是不要迷信GPU。

GPU算力表

参考资源链接:[深度学习配置指南:CUDAcuDNN详解](https://wenku.youkuaiyun.com/doc/6j3oo8yb4n?utm_source=wenku_answer2doc_content) 在Windows 10或Windows 11系统上进行深度学习模型开发时,选择合适的CUDAcuDNN版本至关重要,以确保TensorFlowPyTorch框架的兼容性最佳性能。首先,你应该访问NVIDIA的官方网站获取CUDAcuDNN的最新版本信息,并参考相关硬件的兼容性列表。对于CUDA,选择一个稳定且被广泛支持的版本是很重要的,例如CUDA 11.7,因为它在较新较旧的硬件上都具有良好的支持。对于cuDNN,应该下载与CUDA版本相匹配的版本,并确保你的GPU算力满足该cuDNN版本的最低要求。可以通过NVIDIA控制面板来验证GPU算力。在安过程中,还需要确保你的系统中安了正确版本的Visual Studio,因为CUDA的安某些开发工作可能会用到。对于Visual Studio,推荐使用Visual Studio 2019或2022版本,并确保安了C++开发工具。安CUDAcuDNN后,你可能需要在TensorFlowPyTorch中配置环境变量,以确保框架可以识别到CUDAcuDNN的安路径。TensorFlowPyTorch都有详细的文档说明如何进行这些设置。此外,考虑到版本更新可能会带来新的特性性能改进,建议定期检查并更新CUDAcuDNN以及GPU驱动程序深度学习框架。通过这些步骤,你可以确保你的开发环境既能满足当前的需求,又能适应未来技术的发展。 参考资源链接:[深度学习配置指南:CUDAcuDNN详解](https://wenku.youkuaiyun.com/doc/6j3oo8yb4n?utm_source=wenku_answer2doc_content)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值