操作系统
选择目前主流的版本,windows
10
驱动
到官方网站 https://www.nvidia.cn/geforce/drivers/,根据自己的实际情况,填写显卡类型、产品序列、操作系统、语言和下载类型等信息
比如以显卡 1660 为例
一般情况下,都是下载最新的驱动安装程序,傻瓜式安装即可
cuda
首先到 https://developer.nvidia.com/cuda-downloads? 下载自己需要的版本,如 10.2,下载后,双击 exe
文件开始安装
这是一个临时目录,随意选取一个即可
到此 cuda
就安装好了,接下来我们打开 sample
目录,通过执行其中的 deviceQuery
来检查下安装结果
我们选择用 visual studio 2019
版本打开这个工程,直接编译运行
从结果中可以看出,cuda
已经安装成功了
安装cuDNN
cuDNN
是 nvidia
公司开发的一套用于 DNN
也就是深度神经网络的加速库,下面我们就来安装它。进入 官网 下载安装包,这里需要登录,如果没有账号的话就去注册一个。
下载下来的文件是一个压缩包,解压后有三个文件夹 bin
、include
和 lib
,依次将三个文件夹中的内容拷贝到 cuda
的安装目录(默认情况下是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2
)下的 bin
、include
和 lib
下,这个要一一对应,如下图所示。
至此,cuDNN
也就安装好了。
python环境
这里建议使用 anaconda
,便于进行 python
多版本的管理和使用。从官网下载安装包 https://www.anaconda.com/products/individual,傻瓜式安装,注意一点,将 anaconda
的路径加入到系统环境变量中
pip源
在 C:\Users\$登录用户名
下创建文件夹 pip
,然后在 pip
文件夹下创建文件 pip.ini
,文件内容如下
[global]
index-url = http://mirrors.aliyun.com/pypi/simple/
[install]
trusted-host=mirrors.aliyun.com