0
点赞
收藏
分享

微信扫一扫

OCR识别、模型训练等GPU环境安装配置(windows)

代码敲到深夜 2022-01-07 阅读 36

前提:显卡驱动正常。

第一步:查看本机cuda版本

1、打开命令行

2、输入nvidia-smi

3、查看对应的 cuda version

第二步:下载对应的CUDA并安装

1、CUDA Toolkit 11.1.0 | NVIDIA Developer

如果要选择其他版本,可以利用站内搜索功能进行查找。

 2、下载后安装,自定义安装过程中可以勾选组件,值得注意的是DisplayDriver组件 ,如果当前版本大于新版本,则去掉勾选。

第三步:下载对应的cuDNN

1、cuDNN Archive | NVIDIA Developer

查找自己对应的CUDA版本

 2、下载完成后解压

3、解压后的文件覆盖到CUDA 目录下即可。该示例中默认安装路径(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1)

举报

相关推荐

0 条评论