Shaw0xyz 发表于 2024-7-6 15:03:54

Linux 服务器下非root用户安装CUDA完整流程

本帖最后由 Shaw0xyz 于 2024-7-6 16:11 编辑

1. 引言

在深度学习和科学计算领域,CUDA作为NVIDIA提供的并行计算平台和编程模型被广泛使用。对于普通用户而言,如何在没有root权限的Linux服务器上安装CUDA是一个常见的问题。本文将详细介绍在Linux服务器上以非root用户身份安装CUDA的完整流程。

2. 准备工作

2.1 确认系统需求

安装CUDA之前,需要确认系统满足以下基本需求:
(1) 操作系统:支持的Linux发行版
(2) GPU硬件:NVIDIA GPU
(3) 驱动程序:相应的NVIDIA驱动

2.2 下载CUDA安装包

从NVIDIA官网(https://developer.nvidia.com/cuda-downloads)下载合适的CUDA Toolkit安装包。选择运行文件(.run)版本,因为它可以在没有root权限的情况下安装。

3. 安装CUDA

3.1 创建安装目录

在用户主目录下创建一个目录,用于存放CUDA Toolkit文件。

    mkdir ~/cuda

3.2 运行安装程序

使用下载的CUDA安装包,运行安装程序。

    sh cuda_<version>_linux.run --silent --toolkit --toolkitpath=~/cuda

3.3 配置环境变量

安装完成后,需要配置环境变量,以便系统能够找到CUDA工具和库。

    echo 'export PATH=$HOME/cuda/bin:$PATH' >> ~/.bashrc
    echo 'export LD_LIBRARY_PATH=$HOME/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc

    source ~/.bashrc

4. 验证安装

4.1 检查CUDA版本

使用`nvcc`命令检查CUDA版本,确保安装成功。

    nvcc --version

4.2 运行示例程序

在CUDA安装目录中有许多示例程序,可以编译并运行这些程序以确认安装正确。

    cd ~/cuda/samples/1_Utilities/deviceQuery
    make
    ./deviceQuery

如果显示GPU设备信息,则说明CUDA安装成功。

5. 常见问题解答

5.1 安装过程中遇到权限问题怎么办?

确保在执行安装命令时使用了正确的路径,并且所选目录有写权限。

5.2 如何卸载CUDA?

删除CUDA目录,并从`~/.bashrc`中移除相关环境变量配置。

    rm -rf ~/cuda
    sed -i '/cuda/d' ~/.bashrc

6. 结论

通过以上步骤,用户可以在Linux服务器上以非root身份成功安装和配置CUDA环境。希望本文能帮助您解决实际问题,顺利开展基于CUDA的开发工作。






/ 荔枝学姐de课后专栏 /

Hi!这里是荔枝学姐~

欢迎来到我的课后专栏

自然语言学渣 NLP摆烂姐

热衷于技术写作 IT边角料

AIGC & Coding & Linux ...

~互撩~ TG: @Shaw_0xyz
页: [1]
查看完整版本: Linux 服务器下非root用户安装CUDA完整流程