这个暑假,趁着空闲时学习机器学习,于是想要配置自己电脑的gpu环境。
显卡驱动
查看自己的驱动版本
也可以在程序中查看
对应自己的版本下载cuda
我的驱动是398.27,于是我去官网下载了9.2的版本
https://developer.nvidia.com/cuda-toolkit-archive
然后测试一下,命令行输入
nvcc -V
如果出现
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Tue_Jun_12_23:08:12_Central_Daylight_Time_2018
Cuda compilation tools, release 9.2, V9.2.148
安装成功
下载安装对应CUDNN
在英伟达网站 https://developer.nvidia.com/rdp/cudnn-download 下载与cuda对应的库,需要注册一个账号。
下载之后解压缩,将.h .lib 和.dll 文件分别拷贝到cuda安装目录下对应文件夹下。
至此应该就是安装成功了,至于添加环境什么的,如果你是装在c盘,系统自动会帮你添加,无需自己配置。
下载对应的cudnn
这里我是将解压出来的cuda重命名为cudnn,并复制到cuda文件夹下
将.h .lib 和.dll 文件分别拷贝到cuda安装目录下对应文件夹下。