如何在CentOS上安装PyTorch GPU版
在CentOS上安装PyTorch GPU版,可按以下步骤进行:
- 更新系统:
sudo yum update -y
。 - 安装NVIDIA驱动:从NVIDIA官网下载对应GPU型号的驱动程序并安装,安装过程中可能需要停止图形界面并重启系统。
- 安装CUDA Toolkit:从NVIDIA官网下载适合系统的版本,如使用RPM包安装,需添加仓库并执行安装命令,安装后配置环境变量。
- 安装cuDNN库:从NVIDIA官网下载与CUDA版本匹配的cuDNN库,解压后复制文件到CUDA目录并设置权限。
- 安装PyTorch:使用pip安装,根据CUDA版本选择合适的安装命令,如
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
(CUDA 11.7版本)。 - 验证安装:运行Python脚本
import torch; print(torch.__version__); print(torch.cuda.is_available())
,若输出显示版本号且cuda.is_available()
为True,则安装成功。
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权请联系我们,一经查实立即删除!