cuda toolkit安装centos
时间: 2025-02-09 14:11:50 浏览: 46
### 如何在 CentOS 上安装 CUDA Toolkit
#### 准备工作
确保系统已更新至最新状态并安装必要的依赖项。对于 GCC 编译器,可以通过 `yum` 进行安装[^2]:
```bash
sudo yum install gcc
```
#### 下载 CUDA 工具包
访问 NVIDIA 官方网站获取适用于 CentOS 的 CUDA 版本,并按照指引完成下载过程[^3]。
#### 执行安装脚本
通常情况下,下载后的文件是一个可执行的 `.run` 文件。为了安全起见,在运行之前先给予其执行权限:
```bash
chmod +x cuda_<version>_linux.run
```
接着启动安装向导:
```bash
sudo ./cuda_<version>_linux.run
```
遵循屏幕提示逐步操作即可顺利完成整个流程。值得注意的是,默认路径会设置为 `/usr/local/cuda-<version>` 同时创建指向当前版本号的软链接 `/usr/local/cuda -> /usr/local/cuda-<version>` 以便于后续管理维护[^1].
#### 验证安装成果
通过 Python 测试 GPU 是否可用是一种有效的方式。具体做法是在终端内调用 Python 解释器后加载 PyTorch 库来生成随机数矩阵并尝试将其放置到设备上显示出来:
```python
import torch
x = torch.rand(5, 3).to('cuda')
print(x)
```
当一切正常运作时,上述代码应当返回一个位于 GPU 设备上的张量对象表示形式[^4].
阅读全文
相关推荐


















