问小白 wenxiaobai
资讯
历史
科技
环境与自然
成长
游戏
财经
文学与艺术
美食
健康
家居
文化
情感
汽车
三农
军事
旅行
运动
教育
生活
星座命理

自由安装CUDA【保姆级教程—从卸载到安装指定版本】

创作时间:
作者:
@小白创作中心

自由安装CUDA【保姆级教程—从卸载到安装指定版本】

引用
CSDN
1.
https://m.blog.csdn.net/m0_63324772/article/details/142864078

本文是一个关于如何自由安装CUDA的保姆级教程,从卸载到安装指定版本,手把手教你完成整个过程。

这里以安装CUDA 11.6为例,因为后续将部署YOLOv5,如果需要安装其他版本,可以直接从官方下载。

查看本机已安装的CUDA版本

在命令行中输入以下命令:

nvcc -V

然后输入:

nvidia-smi

查看自己的显卡(GPU)支持最高版本的CUDA是多少。

下载并安装CUDA

下载自己想要的CUDA版本。安装时建议选择“精简”模式,并将软件安装在全英文路径的文件夹中,以避免出现错误。

安装cuDNN

将cuDNN的四个文件复制到CUDA的安装路径下。

配置环境变量

双击Path进入环境变量,然后新建三个环境变量:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\extras\CUPTI\lib64

配置完成后,可以在“应用与功能”中查看是否正确安装。

安装完成后,就可以开始YOLOv5的部署了。

© 2023 北京元石科技有限公司 ◎ 京公网安备 11010802042949号