亲爱的小伙伴们,我来更新了!接着上一小节的内容,在熟悉自己电脑相关配置的基础上,我们安装完成对应显卡驱动,本小节来说一下CUDA下载方法与安装过程中可能遇到的问题及解决方法。
在下载与安装CUDA之前,需要知道CUDA的版本受限于你电脑中系统版本,安装python版本、对应的pytorch版本,还有相应的安装环境。
1.CUDA下载方法
下载地址:developer.nvidia.com/cuda-toolkit-archive
根据自己电脑的相关配置选择对应的操作系统类型、位数,开始下载
下载完之后运行安装,建议安装到默认路径。
2.CUDA安装过程
开始安装,在这个位置要选择自定义安装方式
选择需要安装的组件,这个位置建议不要重复安装,前面安装显卡驱动时已安装的内容,这个位置可以不安装,这里主要是安装CUDA加速。
路径可以保持默认,也可以自己修改,但这个路径一定要记住,因为后面要在这个路径下安装cudnn。
出现这个蓝色的表示你的电脑中没有对应的Visual studio软件,需要下载安装
安装CUDA过程中,出现这个,说明缺少运行环境,需要下载Visual Studio,点击图片中蓝色就可以进入下载连接
接下来选择性安装就可以,不在用图说明,安装结束后,在回到CUDA安装界面。
cudn的安装正常进行next就可以,直到完成!
3.验证CUDA是否安装完成且正确方法
到此,CUDA成功并正确安装到你自己的电脑中,就可以进行对应的下一步,离yolov5的运行环境搭建成功又进了一步!
恳请专业人士指点,希望小伙伴们多多支持!一起努力!