国内从NVIDIA官网下载GPU驱动、CUDA Toolkit、cuDNN通常比较慢,滴滴云提供了这些软件包的内网链接,在滴滴云GPU云服务器上通过内网链接下载会极大加快下载速度。
一、软件包内网链接地址
二、软件包安装方法
下面以某个版本举例说明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/