一、安装Nvidia CUDA工具包:
The NVIDIA® CUDA® Toolkit provides a development environment for creating high performance GPU-accelerated applications.
建议不要安装最新的版本,哪怕是稳定版也不要这么做(你会知道兼容性让你头疼),除非你想尝试一番。
- 下载CUDA工具包
验证GPU是否支持CUDA,且计算能力>=3.5:
https://developer.nvidia.com/cuda-gpus
- 安装CUDA工具包
- 选择安装路径
- 注意勾选自动配置环境变量
二、安装Nvidia cuDNN:
- 下载cuDNN GPU计算加速库,请选择与CUDA对应的版本
NVIDIA cuDNN is a GPU-accelerated library of primitives for deep neural networks.
-
解压并把文件放置安装目录
把解压的cuDNN各个文件夹下的文件,复制到对应的CUDA安装目录下的文件夹里,如下图所示:
image.png
-
检测是否正常:
运行cmd:
nvcc -V
例:
C:\Users\chonger>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:32_Central_Daylight_Time_2017
Cuda compilation tools, release 9.0, V9.0.176
- 大功告成!
网友评论