文章目录 事前准备 CUDA cuDNN Pytorch，torchvision，torchaudio GPU相关 安装CUDA 替换cuDNN Pytorch相关 安装Pytorch 安装torchvision 安装torchaudio 事前准备 CUDA 去到NVIDIA官网上下载CUDA，我这里下载的是10.2版本的CUDA。下载地址 cuDNN cuDNN的版本要与CUDA版本相对应。下载地址。 cuDNN需要注册账户才可以下载。 Pytorch，torchvision，torchaudio 因为在国内，故直接使用在线pip下载不太好使，因此这里给出一个网站，下载离线whl包进行安装。大集合下载地址 注意，这里所下载的版本也须与CUDA的版本相对应，具体可去Pytorch官网看，我这里给出CUDA10.2的下载地址。
本文介绍了如何配置基于GPU的深度学习环境，详细讲解了从下载CUDA 10.2、cuDNN到安装Pytorch、torchvision和torchaudio的步骤。包括CUDA的自定义安装、cuDNN文件的替换，以及各个库的离线安装和验证过程。
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
10万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
