Pytorch的安装(Cuda+Cudnn)GPU版本
cuda安装之前,需要查看本机的gpu版本
查看自己电脑是否安装了显卡驱动
在桌面右击,打开NVIDIA 控制面板,点击系统信息,我们可以看到自己驱动程序版本,如果没有gpu驱动,建议先安装NVIDIA驱动
点击组件查看支持cuda的版本号,在这支持cuda 11.6.1
安装cuda https://developer.nvidia.com/cuda-toolkit-archive
根据自己的机型
双击安装,直接一路下去,记得把加入环境变量勾上
为了检验我们cuda是否真正安装成功,我们打开命令行窗口(cmd),因为cuda安装过程中,已经自动配置好了环境变量,我们在命令行中输入
nvcc -V
下载cundnn,可选
网址Log in, https://developer.nvidia.com/cudnn-downloads,进入网页,注意先注册
可以使用下面网址,查看适配的 cuDNN,https://developer.nvidia.com/rdp/cudnn-archive
解压文件夹将其复制到我们下载的cuda目录下,一般都是在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA之下
给cudnn添加环境变量
添加上两条环境变量
测试是否安装成功
在cmd运行【C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\extras\demo_suite】路径下的bandwidthTest.exe和deviceQuery.exe,如果都能运行成功,则证明cudnn安装成功。
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\extras\demo_suite
bandwidthTest.exe
deviceQuery.exe
参考 https://blog.csdn.net/weixin_46668520/article/details/134102997
分类:
python
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· winform 绘制太阳,地球,月球 运作规律
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· AI与.NET技术实操系列(五):向量存储与相似性搜索在 .NET 中的实现
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理