如何在GCP上的Linux实例中直接从Nvidia网站下载cuDNN?

11
我想在Google云平台上的Linux机器上安装tensorflow-gpu。我没有使用GCP提供的深度学习VM,所以我在我的Linux实例上安装了Anaconda,现在我想安装tensorflow。我已经安装了NVIDIA驱动程序和CUDA。它们可以直接下载到云实例中。但是对于cuDNN,我必须将其下载到本地计算机,然后将其上传到云实例。有没有办法直接从NVIDIA网站下载该文件到我的云实例?谢谢。
 CUDNN_URL="developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz" 
 wget -c ${CUDNN_URL} 

使用以下命令,我们可以直接下载cudnnv5.1,并且我也看到了版本6.5的链接。我尝试使用相同的链接并输入我想要的版本,但无效。请问有没有办法使用CUDNN_URL通过wget或curl直接下载cudnn v7.1或更高版本,而不需要登录Nvidia账户?

1
你链接中缺少的是完整版本。它是在7.2.1版本中引入的。 7.2.1及更高版本:developer.download.nvidia.com/compute/redist/cudnn/v7.2.1/cudnn-9.2-linux-x64-v7.2.1.38.tgz 之前的版本为:developer.download.nvidia.com/compute/redist/cudnn/v7.1.4/cudnn-9.0-linux-x64-v7.1.tgz - robertm.tum
请查看下面我的答案,以获得完整的解释。 - robertm.tum
4个回答

13

cuDNN归档命名规则发生了变化。

自7.2.1版本起,NVIDIA在归档文件名中添加了完整的版本号,而不是以前使用的简短版本号。

这意味着7.2.1的下载链接为: https://developer.download.nvidia.com/compute/redist/cudnn/v7.2.1/cudnn-9.2-linux-x64-v7.2.1.38.tgz 而不是: https://developer.download.nvidia.com/compute/redist/cudnn/v7.2.1/cudnn-9.2-linux-x64-v7.2.tgz

您可以按照这个模式进行操作:

VERSION_FULL="8.1.0.77"
VERSION="${VERSION_FULL%.*}"
CUDA_VERSION="11.2"
OS_ARCH="linux-x64"
CUDNN_URL="https://developer.download.nvidia.com/compute/redist/cudnn/v${VERSION}/cudnn-${CUDA_VERSION}-${OS_ARCH}-v${VERSION_FULL}.tgz" 
wget -c ${CUDNN_URL}

生成的链接将是:https://developer.download.nvidia.com/compute/redist/cudnn/v8.1.0/cudnn-10.2-linux-x64-v8.1.0.77.tgz


3
您,先生,真是个大师! - Momchill
以防万一,这是我为Windows准备的最新版本:https://developer.nvidia.com/compute/cudnn/secure/8.3.1/local_installers/11.5/cudnn-windows-x86_64-8.3.1.22_cuda11.5-archive.zip - o_nix

2

是的,我就是用scp命令上传了cuda和驱动程序。我想我已经得到了我想要的答案。谢谢。 - Gayal Kuruppu
有直接的链接,但并非始终公开可用。 - robertm.tum

2

编辑:此方法适用于CUDNN v8.9及以下版本,对于v8.9+版本不再适用。

更简单(也更好)的方法:

只需浏览https://developer.download.nvidia.com/compute/redist/cudnn/并获取链接(源码或安装程序包)。不同版本的URL模式并不是非常一致。

例如,如果您需要在Linux x86_64上使用CUDNN v8.7:


1
这在8.9及以后版本不适用。 - Shital Shah
有没有仅限命令行的方法适用于较新版本? - ahron

0

如果你真的关心(我也是)将数据下载到本地机器,然后上传到GCP实例中的cuda和cudnn文件,那么你可以在很短的时间内为你的GCP实例设置一个GUI。查看这个https://www.youtube.com/watch?v=e3RnnmcNI_E或任何VNC服务器教程。之后,你可以直接使用Web浏览器下载任何文件。


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接