本文介绍了在CentOS7系统中如何安装PyTorch，强调了驱动版本、CUDA Toolkit与PyTorch版本的匹配性。首先，通过命令检查驱动版本（520.61）和CUDA版本（11.8）。接着，提供了安装CUDA Toolkit的参考链接。当CUDA版本与驱动不匹配时，提出了降级CUDA或升级驱动的解决方案。最后，详细指导了如何从PyTorch官网选择并离线下载适配的PyTorch版本，特别是对于Python和CUDA版本的注意事项。
1.准备工作
驱动版本（Driver Version）、CUDA Toolkit（CUDA Version）、PyTorch版本要相互匹配才能成功使用.
2.查看驱动版本
使用命令查看驱动版本， 如果没有该命令请先安装驱动参考:
CentOS7安装NVIDIA GPU驱动程序和CUDA工具包-优快云博客
nvidia-smi
可见我当前的驱动版本是520.61，CUDA版本是11.8
安装cuda toolkit 参考：
3.CUDA版本和驱动版本不匹配的应对方法
如果现在你的CUDA版本和驱动版本不匹配，就可以：
- 方法1：使用低版本CUDA Toolkit
- 方法2：更新驱动升级为更高版本
4.安装pytorch
登录官网选择合适版本
如果想选择其他版本请点击下面previous versions of pytorch
离线下载
如果是离线下载在其他服务器上部署，出现python版本和pytorch版本不对应的情况
打开后面的下载地址 手动下载选择合适的包：
https://download.pytorch.org/whl/cu118
2.1.2是指pytorch的版本 、cu118是 cuda的版本、 cp311代表python3.11
扫一扫
2021
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
