我要在windows 运行pytorch
下载时候,需要cuda的版本
一、查看电脑上有没有cuda
二、下载cuda
https://blog.csdn.net/qq_40968179/article/details/128996692
1. 打开英伟达控制面板
2. 帮助
3. 得到nvidia能驱动的最大版本
4. 安装cuda 【上面文档】
踩坑1:
5. 安装cudnn
这两个是一起的。
深度学习本质上就是训练深度卷积神经网络。尽管我们已经可以用cuda使得显卡来完成并行计算任务,但是所有的操作还是比较底层的、复杂的。是否在cuda之上有一个专门用于深度神经网络的SDK库来加速完成相关特定的深度学习操作,答案就是cudnn
cuDNN是用于深度神经网络的GPU加速库
https://developer.nvidia.com/cudnn
下载之后,解压缩,将CUDNN压缩包里面的bin、include、lib文件直接复制到CUDA的安装目录下,直接覆盖安装即可完成
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 全程不用写代码,我用AI程序员写了一个飞机大战
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!