Ubuntu环境下的cuda安装

在Ubuntu环境下安装CUDA,可以按照以下步骤进行:

  1. 下载CUDA Toolkit:从NVIDIA官方网站下载适用于Ubuntu的CUDA Toolkit安装包。

  2. 关闭图形界面:在安装CUDA前,最好在命令行界面下执行操作,可以通过按下 Ctrl + Alt + F1切换到命令行终端。

  3. 安装依赖:在命令行终端中运行以下命令安装CUDA所需的依赖库:

    sudo apt update
    sudo apt install build-essential
     
     
  4. 运行安装脚本:在命令行终端中进入下载的CUDA Toolkit安装包所在目录,并执行以下命令运行安装脚本:

    chmod +x cuda_*.run
    sudo ./cuda_*.run
     
     

    注意,*.run需要替换为下载的CUDA Toolkit安装包的版本号。

  5. 安装过程中会提示同意许可协议、选择安装位置和是否安装NVIDIA显卡驱动等选项,根据实际需求进行选择。

  6. 设置环境变量:安装完成后,需要将CUDA的路径添加到环境变量中。在 ~/.bashrc文件中添加以下内容:

    export PATH=/usr/local/cuda-X.X/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-X.X/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
     
     

    注意,X.X需要替换为安装的CUDA版本号。

  7. 更新环境变量:执行以下命令使环境变量生效:

    source ~/.bashrc
     
     
  8. 验证安装:运行以下命令查看CUDA版本:

    nvcc --version
     
     

    运行以下命令查看NVIDIA驱动版本:

    nvidia-smi
     
     

安装完成后,CUDA就成功安装在Ubuntu系统中了,你可以开始使用CUDA来进行GPU加速的计算

posted @ 2025-03-17 11:01  yu8yu7  阅读(832)  评论(0)    收藏  举报