深度学习环境GPU配置
操作系统为 Windows10。 独显为英伟达显卡。
安装cuda
查看显卡是否支持cuda, 以及对应cuda版本。
打开NVIDIA控制面板, 左下角系统信息, 查看组件中cuda驱动版本。 安装的cuda不能高于此版本(可以低, 比如11.3.101 可以装cuda10.2)。
顺便
去官网下载对应版本。
安装过程中注意选择自定义, 自定义组件中取消 VS。
C盘空间够的话尽量默认安在C盘, 记住默认路径。
安装完成后配置系统变量, 在path下添加
1 |
|
打开命令行输入 nvcc -V
检查是否安装成功。
安装cudnn
查询cuda对应的cudnn版本, 去官网下载, 需要注册登录以及填一些信息和问卷。
下载压缩包后解压, 将三个文件夹覆盖到cuda安装目录即可。(可以将文件夹的文件复制到cuda安装目录的对应文件夹中)。