虽然paddle的训练器支持CPU,但如果用CPU的话,两三万张图片训练时间会长达三四天。
所以为了缩短时间,我们要采取GPU(显卡)来进行训练,这就需要英伟达的CUDA了
1、下载cuda9.0
下载链接:https://developer.nvidia.com/cuda-toolkit-archive
2、安装cuda
安装cuda时,第一次会让设置临时解压目录,第二次会让设置安装目录;
临时解压路径,建议默认即可,也可以自定义。安装结束后,临时解压文件夹会自动删除;
安装目录,建议默认即可;
注意:临时解压目录千万不要和cuda的安装路径设置成一样的,否则安装结束,会找不到安装目录的!!!
选择自定义安装
安装完成后,配置cuda的环境变量;
命令行中,测试是否安装成功;
步骤如下(部分过程图):
安装完成,先查看系统变量,然后添加cuda的系统变量:
查看:
添加系统变量后如下图所示:
测试:
win+R
nvcc -V
nvcc --version
显示cuda版本号9.0 V9.0.176
说明安装成功!好,就是这样!