本文指导在Ubuntu 20.04上安装GPU加速的PyTorch，首先需从NVIDIA官网下载并安装显卡驱动，然后安装CUDA和cuDNN库。参照提供的教程，经过实践验证，流程包括Nvidia驱动、CUDA和cuDNN的详细安装步骤，解决报错的方法，并提供了额外的资源链接，以帮助完成PyTorch的安装。
昨天晚上花了几个小时，终于成功在ubuntu20.04安装了GPU加速版本的PyTorch，过程中发现网上现有资料比较分散，对新手不友好，所以想做一篇整合文章分享：
首先要有一台带GPU的机器，并从NVIDIA官网下载安装显卡驱动（参考资料如下）:
如何在Ubuntu 20.04 LTS上安装Nvidia驱动程序
ubuntu20.04系统用.run文件安装nvidia显卡驱动
在安装完显卡驱动后，还需要安装CUDA（Compute Unified Device Architecture），CUDA是NVIDIA推出的通用并行计算架构，可以使深度学习框架调用GPU来解决问题。另外，通常我们还需要安装cuDNN库，cuDNN库是由NVIDIA开发的专门用于深度神经网络的GPU加速库（参考资料如下）：
参考以上三篇大佬文章，亲测有效，如果实操过程中遇到报错，可以百度以下哈，作者也是历经千辛万苦才弄好的，另外在找资料的过程中，我还有意外收获（参考资料见下）：
深度学习三件套:Ubuntu 20.04 安装 NVIDIA 驱动/CUDA/cuDNN全流程
最后，再安装一下pytorch即可（其中的“×”表示所需要的版本，请自行查询）：
最低0.47元/天 解锁文章
扫一扫
627
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
