Linux的cuda安装
时间: 2025-04-19 11:49:11 AIGC 浏览: 57
### 如何在Linux系统上安装CUDA
#### 安装准备
为了顺利安装CUDA,在开始之前需确认已满足一些基本条件。这包括拥有支持CUDA技术的NVIDIA显卡以及对应的驱动程序已经正确安装[^1]。
#### 下载CUDA Toolkit
对于非root用户的安装情况,可以从NVIDIA官方网站获取适用于Linux系统的CUDA工具包离线安装档。选择版本时应考虑与现有环境兼容的因素,比如操作系统版本和所使用的GCC编译器版本等。
#### 设置安装路径
创建一个用于存放CUDA软件包的目标目录,例如`$HOME/cuda`。此操作可以通过命令行完成:
```bash
mkdir -p $HOME/cuda
```
#### 解压缩并配置环境变量
下载完成后得到`.run`格式的安装文件,通过指定目标路径来执行该脚本实现静默安装;如果下载的是tgz归档,则先将其解压到上述建立好的个人空间内:
```bash
tar --directory=$HOME/cuda -zxvf cuda-linux-x64-runfile-*.tgz
```
接着编辑shell初始化文件(如`.bashrc`),加入如下两行以更新PATH和LD_LIBRARY_PATH环境变量指向新的CUDA库位置:
```bash
export PATH=$HOME/cuda/bin:$PATH
export LD_LIBRARY_PATH=$HOME/cuda/lib64:$LD_LIBRARY_PATH
```
使更改生效可运行source指令加载修改后的设置:
```bash
source ~/.bashrc
```
#### 验证安装成功与否
最后一步是验证CUDA是否被正确安装。可以尝试编译官方提供的样例项目或者编写简单的测试代码调用cuInit()函数来进行初步检测[^2]。
```cpp
#include <stdio.h>
#include <cuda_runtime.h>
int main(){
int deviceCount;
cudaError_t error = cudaGetDeviceCount(&deviceCount);
if(error != cudaSuccess){
printf("CUDA Error: %s\n", cudaGetErrorString(error));
return 1;
}
printf("Number of CUDA devices: %d\n", deviceCount);
}
```
编译上面这段C++源码需要用到nvcc编译器,并且链接必要的CUDA runtime library:
```bash
nvcc test_cuda.cu -o test_cuda
./test_cuda
```
当看到输出显示可用设备数量而非错误信息即表示安装过程顺利完成。
阅读全文
相关推荐



















