国内从NVIDIA官网下载GPU驱动、CUDA Toolkit、cuDNN通常比较慢,下面提供了这些软件包的内网链接会极大加快下载速度。
一、软件包内网链接地址
CUDA 是英伟达显卡进行各种高性能运算必备的库,不同软件包依赖的 CUDA 版本可能不同,你需要根据你所使用的深度学习框架选择合适的 CUDA 和 cuDNN 版本。 版本 | Windows 10 | Linux |
---|
CUDA 10.2 | cuda_10.2.89_441.22_win10.exe | cuda_10.2.89_440.33.01_linux.run | CUDA 10.1 | cuda_10.1.243_426.00_win10.exe | cuda_10.1.243_418.87.00_linux.run | CUDA 10.0 | cuda_10.0.130_411.31_win10.exe | cuda_10.0.130_410.48_linux.run | CUDA 9.2 | cuda_9.2.148_win10.exe | cuda_9.2.148_396.37_linux.run | CUDA 9.1 | cuda_9.1.85_win10.exe | cuda_9.1.85_387.26_linux.run | CUDA 9.0 | cuda_9.0.176_win10.exe | cuda_9.0.176_384.81_linux.run | | 以下版本较低,推荐使用上面的版本。 | | CUDA 8.0 GA2 | cuda_8.0.61_win10.exe | cuda_8.0.61_375.26_linux.run | CUDA 8.0 GA1 | cuda_8.0.44_win10.exe | cuda_8.0.44_linux.run | CUDA 7.5 | cuda_7.5.18_win10.exe | cuda_7.5.18_linux.run | CUDA 7.0 | cuda_7.0.28_windows.exe | cuda_7.0.28_linux.run | CUDA 6.5 | cuda_6.5.14_windows_general_64.exe | cuda_6.5.14_linux_64.run |
cuDNN cuDNN 下载需要注册,地址如下: - 最新版:https://developer.nvidia.com/cudnn
- 旧版:https://developer.nvidia.com/rdp/cudnn-archive
cuDNN 是英伟达推出的专门用于深度学习加速计算的库,一般来说比使用纯 CUDA 速度要快不少。比如 Keras 里有普通的 LSTM 以及 CuDNNLSTM,速度相差最高有十倍。如果没有设计特殊的结构,cuDNN 应该是你的首选。 你需要根据你所使用的深度学习框架选择合适的 CUDA 和 cuDNN 版本。 cuDNN | CUDA9.0 | CUDA9.2 | CUDA10.0 | CUDA10.1 |
---|
7.6.1 | cudnn-9.0-linux-x64-v7.6.1.34.tgz (392.82MB) | cudnn-9.2-linux-x64-v7.6.1.34.tgz (396.89MB) | cudnn-10.0-linux-x64-v7.6.1.34.tgz (460.88MB) | cudnn-10.1-linux-x64-v7.6.1.34.tgz (497.63MB) | 7.6.0 | cudnn-9.0-linux-x64-v7.6.0.64.tgz (374.59MB) | cudnn-9.2-linux-x64-v7.6.0.64.tgz (378.84MB) | cudnn-10.0-linux-x64-v7.6.0.64.tgz (440.90MB) | cudnn-10.1-linux-x64-v7.6.0.64.tgz (477.61MB) | 7.5.1 | cudnn-9.0-linux-x64-v7.5.1.10.tgz (357.13MB) | cudnn-9.2-linux-x64-v7.5.1.10.tgz (361.63MB) | cudnn-10.0-linux-x64-v7.5.1.10.tgz (412.76MB) | cudnn-10.1-linux-x64-v7.5.1.10.tgz (449.47MB) | 7.5.0 | cudnn-9.0-linux-x64-v7.5.0.56.tgz (357.11MB) | cudnn-9.2-linux-x64-v7.5.0.56.tgz (361.61MB) | cudnn-10.0-linux-x64-v7.5.0.56.tgz (412.76MB) | cudnn-10.1-linux-x64-v7.5.0.56.tgz (449.46MB) | 7.4.2 | cudnn-9.0-linux-x64-v7.4.2.24.tgz (349.00MB) | cudnn-9.2-linux-x64-v7.4.2.24.tgz (353.84MB) | cudnn-10.0-linux-x64-v7.4.2.24.tgz (404.82MB) | - | 7.4.1 | cudnn-9.0-linux-x64-v7.4.1.5.tgz (347.17MB) | cudnn-9.2-linux-x64-v7.4.1.5.tgz (352.02MB) | cudnn-10.0-linux-x64-v7.4.1.5.tgz (403.12MB) | - | 7.3.1 | cudnn-9.0-linux-x64-v7.3.1.20.tgz (344.85MB) | cudnn-9.2-linux-x64-v7.3.1.20.tgz (351.05MB) | cudnn-10.0-linux-x64-v7.3.1.20.tgz (400.58MB) | - | 7.3.0 | cudnn-9.0-linux-x64-v7.3.0.29.tgz (340.43MB) | - | cudnn-10.0-linux-x64-v7.3.0.29.tgz (396.06MB) | - | 7.2.1 | cudnn-9.0-linux-x64-v7.2.1.38.tgz (331.19MB) | cudnn-9.2-linux-x64-v7.2.1.38.tgz (337.08MB) | - | - | 7.1.4 | cudnn-9.0-linux-x64-v7.1.tgz (390.26MB) | cudnn-9.2-linux-x64-v7.1.tgz (401.58MB) | - | - |
以下版本较低,推荐使用上面的版本。 cuDNN | CUDA6.5 | CUDA7.0 | CUDA7.5 | CUDA8.0 | CUDA9.0 |
---|
7.2.1 | - | - | - | cudnn-8.0-linux-x64-v7.2.1.38.tgz (189.78MB) | cudnn-9.0-linux-x64-v7.2.1.38.tgz (331.19MB) | 7.1.4 | - | - | - | - | cudnn-9.0-linux-x64-v7.1.tgz (390.26MB) | 7.1.3 | - | - | - | cudnn-8.0-linux-x64-v7.1.tgz (245.96MB) | cudnn-9.0-linux-x64-v7.1.tgz (387.77MB) | 7.1.2 | - | - | - | cudnn-8.0-linux-x64-v7.1.tgz (246.13MB) | cudnn-9.0-linux-x64-v7.1.tgz (387.42MB) | 7.0.5 | - | - | - | cudnn-8.0-linux-x64-v7.tgz (267.98MB) | cudnn-9.0-linux-x64-v7.tgz (332.66MB) | 7.0.4 | - | - | - | cudnn-8.0-linux-x64-v7.tgz (267.98MB) | cudnn-9.0-linux-x64-v7.tgz (332.67MB) | 6.0 | - | - | cudnn-7.5-linux-x64-v6.0.tgz (109.43MB) | cudnn-8.0-linux-x64-v6.0.tgz (191.82MB) | - | 5.1 | - | - | cudnn-7.5-linux-x64-v5.1.tgz (77.18MB) | cudnn-8.0-linux-x64-v5.1.tgz (98.39MB) | - | 5 | - | - | cudnn-7.5-linux-x64-v5.0-ga.tgz (75.66MB) | cudnn-8.0-linux-x64-v5.0-ga.tgz (94.58MB) | - | 4 | - | cudnn-7.0-linux-x64-v4.0-prod.tgz (81.73MB) | - | - | - | 3 | - | cudnn-7.0-linux-x64-v3.0-prod.tgz (63.81MB) | - | - | - | 2 | cudnn-6.5-linux-x64-v2.tgz (15.18MB) | - | - | - | - |
下面是 Windows 10 的各个版本的安装包: cuDNN | CUDA9.0 | CUDA9.2 | CUDA10.0 | CUDA10.1 |
---|
7.6.1 | cudnn-9.0-windows10-x64-v7.6.1.34.zip (201.80MB) | cudnn-9.2-windows10-x64-v7.6.1.34.zip (203.54MB) | cudnn-10.0-windows10-x64-v7.6.1.34.zip (237.73MB) | cudnn-10.1-windows10-x64-v7.6.1.34.zip (263.75MB) | 7.6.0 | cudnn-9.0-windows10-x64-v7.6.0.64.zip (192.70MB) | cudnn-9.2-windows10-x64-v7.6.0.64.zip (194.52MB) | cudnn-10.0-windows10-x64-v7.6.0.64.zip (227.76MB) | cudnn-10.1-windows10-x64-v7.6.0.64.zip (253.77MB) | 7.5.1 | cudnn-9.0-windows10-x64-v7.5.1.10.zip (184.06MB) | cudnn-9.2-windows10-x64-v7.5.1.10.zip (185.99MB) | cudnn-10.0-windows10-x64-v7.5.1.10.zip (213.77MB) | cudnn-10.1-windows10-x64-v7.5.1.10.zip (239.76MB) | 7.5.0 | cudnn-9.0-windows10-x64-v7.5.0.56.zip (184.05MB) | cudnn-9.2-windows10-x64-v7.5.0.56.zip (185.98MB) | cudnn-10.0-windows10-x64-v7.5.0.56.zip (213.78MB) | cudnn-10.1-windows10-x64-v7.5.0.56.zip (239.75MB) | 7.4.2 | cudnn-9.0-windows10-x64-v7.4.2.24.zip (180.04MB) | cudnn-9.2-windows10-x64-v7.4.2.24.zip (182.13MB) | cudnn-10.0-windows10-x64-v7.4.2.24.zip (209.84MB) | - | 7.4.1 | cudnn-9.0-windows10-x64-v7.4.1.5.zip (179.09MB) | cudnn-9.2-windows10-x64-v7.4.1.5.zip (181.16MB) | cudnn-10.0-windows10-x64-v7.4.1.5.zip (208.93MB) | - | 7.3.1 | cudnn-9.0-windows10-x64-v7.3.1.20.zip (177.85MB) | cudnn-9.2-windows10-x64-v7.3.1.20.zip (180.61MB) | cudnn-10.0-windows10-x64-v7.3.1.20.zip (207.59MB) | - | 7.3.0 | cudnn-9.0-windows10-x64-v7.3.0.29.zip (176.12MB) | - | cudnn-10.0-windows10-x64-v7.3.0.29.zip (205.84MB) | - | 7.2.1 | cudnn-9.0-windows10-x64-v7.2.1.38.zip (171.64MB) | cudnn-9.2-windows10-x64-v7.2.1.38.zip (174.26MB) | - | - | 7.1.4 | cudnn-9.0-windows10-x64-v7.1.zip (201.39MB) | cudnn-9.2-windows10-x64-v7.1.zip (206.81MB) | - | - |
cuDNN | CUDA6.5 | CUDA7.0 | CUDA7.5 | CUDA8.0 | CUDA9.0 |
---|
7.2.1 | - | - | - | cudnn-8.0-windows10-x64-v7.2.1.38.zip (100.06MB) | cudnn-9.0-windows10-x64-v7.2.1.38.zip (171.64MB) | 7.1.4 | - | - | - | - | cudnn-9.0-windows10-x64-v7.1.zip (201.39MB) | 7.1.3 | - | - | - | cudnn-8.0-windows10-x64-v7.1.zip (128.37MB) | cudnn-9.0-windows10-x64-v7.1.zip (200.17MB) | 7.1.2 | - | - | - | cudnn-8.0-windows10-x64-v7.1.zip (128.47MB) | cudnn-9.0-windows10-x64-v7.1.zip (200.00MB) | 7.0.5 | - | - | - | cudnn-8.0-windows10-x64-v7.zip (139.68MB) | cudnn-9.0-windows10-x64-v7.zip (172.86MB) | 7.0.4 | - | - | - | cudnn-8.0-windows10-x64-v7.zip (139.72MB) | cudnn-9.0-windows10-x64-v7.zip (172.86MB) | 6.0 | - | - | cudnn-7.5-windows10-x64-v6.0.zip (56.01MB) | cudnn-8.0-windows10-x64-v6.0.zip (101.57MB) | - | 5.1 | - | - | cudnn-7.5-windows10-x64-v5.1.zip (40.15MB) | cudnn-8.0-windows10-x64-v5.1.zip (53.95MB) | - | 5 | - | - | cudnn-7.5-windows10-x64-v5.0-ga.zip (39.41MB) | cudnn-8.0-windows10-x64-v5.0-ga.zip (51.21MB) | - |
| | |
---|
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
| | |
二、软件包安装方法
下面以某个版本举例说明NVIDIA GPU驱动、CUDA Toolkit、cuDNN的安装方法。
2.1 安装 GPU 驱动
apt install -y gcc g++ build-essential linux-headers-$(uname -r)
wget http://mirrors.intra.didiyun.com/gpu_resource/NVIDIA-Linux-x86_64-430.34.run
chmod 755 NVIDIA-Linux-x86_64-430.34.run
./NVIDIA-Linux-x86_64-430.34.run
nvidia-smi -pm 1
2.2 安装 CUDA Toolkit
wget http://mirrors.intra.didiyun.com/gpu_resource/cuda_10.1.105_418.39_linux.run
chmod 755 cuda_10.1.105_418.39_linux.run
./cuda_10.1.105_418.39_linux.run
注意上一步已经安装了 GPU 驱动,这一步安装时要跳过驱动安装。
2.3 安装 cuDNN:
wget http://mirrors.intra.didiyun.com/gpu_resource/cudnn-10.1-linux-x64-v7.6.1.34.tgz
tar zxvf cudnn-10.1-linux-x64-v7.6.1.34.tgz -C /usr/local/