由于学习需要,手上有一台图形工作站,GPU是NVIDIA Quarter K2000的,就试着看看装个GPU环境训练
1 安装前准备工作
在官网可以查到自己的GPU是否支持CUDA加速
一般算力在3.0的才可以支持cudnn加速,如果自己的显卡版本没在这上面,还是老老实实使用CPU训练吧(不过一般N卡好像都支持cuda)
2 所需的软件
- CUDA Tookit 9.0
- cuDNN 7.4.1
注意:现在最新的CUDA是10.0,不过最好还是装9.0一下的吧,本人装10.0后会出现问题,另外,cuDnn的版本一定要跟CUDA的版本对应着,一定,一定。
3 下载安装
3.1 安装CUDA
官网根据个人电脑配置自行下载对应的CUDA Toolkit
下载完成后是个exe文件,双击执行就行了,跟普通软件安装方法一样,
中间选择的地方选择 精简安装