1.CUDA安装(有N卡,v10.0)
什么是CUDA
CUDA(ComputeUnified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。
下载安装包
注意:tensorflow2.0对应的cuda版本是10.0,绝对不能下载错。下载网页链接
下载选项:
安装步骤
下载好后打开安装程序,安装步骤如下:
设置如下
点击下一步开始安装。
之后就cuda就算安装成功了。
2.安装cuDNN(v7.5.0)
什么是cuDNN
NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中,如谷歌的Tensorflow、加州大学伯克利分校的流行caffe软件。CUDA看作是一个工作台,而cuDNN就相当于工作的工具。想要在CUDA上运行深度神经网络,就要安装cuDNN。
下载安装包
- 进入官网后,免费注册一个账号后就可以开始下载。
Download cuDNN v7.5.0 (Feb 21, 2019), for CUDA 10.0
解压
下载好后,解压得到名为cuda的文件夹,然后将解压缩后的文件复制到cuda安装路径下的cudnn文件夹(这个文件夹需要自己建)。如图所示:
环境变量配置。
在系统变量的Path里面添加下面的四个项目:缺一不可,且顺序要在最上面(为了避免与其他版本冲突):