优秀的编程知识分享平台

网站首页 > 技术文章 正文

Python机器学习之CUDA环境部署(cuda10.2 pytorch)

nanyue 2024-08-08 19:10:18 技术文章 12 ℃

(前提是要有支持CUDA的显卡,基本都是英伟达公司推出的NVIDIA显卡,具体型号需要到官网上查询)

概念:

CUDACompute Unified Device Architecture,统一计算架构)是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。

cuDNN 是基于CUDA的用于深度神经网络的GPU加速库,有了它才能在GPU上完成深度学习的计算。cuDNN是插入式设计,即所谓安装只需把cuDNN文件复制到CUDA对应文件夹就可以了。

安装步骤:

1.官网下载CUDA

查看本机支持的CUDA版本

英伟达CUDA Toolkit历史工具包的网址:
https://developer.nvidia.com/cuda-toolkit-archive

2.官网下载

cuDNN官网

https://developer.nvidia.com/cudnn

注意与CUDA版本相匹配!

(需要有nvidia账号)

3.安装程序

选择自定义

全选

安装位置建议保持默认位置,并保存位置图片

安装结果

(之前版本安装还得手动设置环境变量,现在不用了,直接测试:nvcc -V

4.安装cuDNN

解压下载好的压缩包,得到cudnn-windows-x86_64-8.9.5.29_cuda12-archive文件夹:

将cudnn文件中对应的文件移动到cuda对用安装目录中,注意不要移动到错误的位置

测试一下CUDA和cudnn是否安装成功

进入目录

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.0\extras\demo_suite

执行上面两个程序会出现以下结果,表示成功

以上就是CUDA部署过程,欢迎批评指正,感谢点赞加关注!

最近发表
标签列表