软工划水日报-CUDA下载与安装 4/17

虽然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

说明安装成功!好,就是这样!

原文地址:https://www.cnblogs.com/Sakuraba/p/14909907.html