centos7 安装cuda toolkit

本文档介绍了如何在CentOS7上安装CUDA Toolkit。首先,查看本机NVIDIA驱动版本,然后根据官方对应关系选择合适的CUDA版本进行下载安装。最后,通过执行验证命令确认CUDA安装成功。
PyTorch 2.6

PyTorch 2.6

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

1.查看本机驱动版本

nvidia-smi

如果没有该命令--安装驱动请参考:CentOS7安装NVIDIA GPU驱动程序和CUDA工具包-优快云博客

我的驱动版本是520.61.05,参考cuda工具包和驱动版本对应关系官网地址:

1. CUDA 12.4 Release Notes — Release Notes 12.4 documentation

CUDA Toolkit Archive | NVIDIA Developer

2.下载安装

选择自己对应的版本进行下载安装

替换成自己下载包的版本和名字

sudo chmod +x cuda_xxx.xx.xx_linux.run
sudo ./cuda_xxx.xx.xx_linux.run

3.验证

使用命名进行验证是否安装成功

nvcc -V 

您可能感兴趣的与本文相关的镜像

PyTorch 2.6

PyTorch 2.6

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

CentOS系统上安装CUDA Toolkit可以按照以下步骤进行: ### 1. 系统准备 确保系统已经安装了合适的内核头文件和开发包,可使用以下命令安装: ```bash sudo yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r) ``` ### 2. 禁用 Nouveau 驱动 Nouveau 是 NVIDIA 显卡的开源驱动,会与 CUDA 所需的 NVIDIA 专有驱动冲突,需要将其禁用。 创建一个新的配置文件: ```bash sudo vim /etc/modprobe.d/blacklist-nouveau.conf ``` 在文件中添加以下内容: ```plaintext blacklist nouveau options nouveau modeset=0 ``` 更新 initramfs : ```bash sudo dracut --force ``` ### 3. 下载 CUDA Toolkit 访问 NVIDIA 官方 CUDA Toolkit 下载页面(https://developer.nvidia.com/cuda-downloads ),选择 CentOS 系统对应的版本进行下载。假设下载的是 CUDA 11.7 的 runfile 安装包,会得到一个类似 `cuda_11.7.0_515.43.04_linux.run` 的文件。 ### 4. 重启系统进入运行级别 3 重启系统并进入运行级别 3 以避免图形界面的干扰: ```bash sudo systemctl set-default multi-user.target sudo reboot ``` ### 5. 安装 CUDA Toolkit 运行下载的 runfile 安装包: ```bash sudo sh cuda_11.7.0_515.43.04_linux.run ``` 安装过程中,按照提示进行操作,接受许可协议,选择要安装的组件(通常选择安装 NVIDIA 驱动、CUDA 工具包和 CUDA 示例)。 ### 6. 配置环境变量 安装完成后,需要配置环境变量。打开 `.bashrc` 文件: ```bash vim ~/.bashrc ``` 在文件末尾添加以下内容: ```plaintext export PATH=/usr/local/cuda/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ``` 使配置生效: ```bash source ~/.bashrc ``` ### 7. 验证安装 使用以下命令验证 CUDA 是否安装成功: ```bash nvcc --version ``` 如果显示 CUDA 版本信息,则说明安装成功。 ### 8. 恢复图形界面 如果需要恢复图形界面,可执行以下命令: ```bash sudo systemctl set-default graphical.target sudo reboot ```
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值