【ubuntu】安装cuda

本文提供了一种详细的步骤来安装和配置CUDA环境。从卸载旧版驱动开始,到下载最新的CUDA驱动,再到安装CUDA Toolkit和设置环境变量,每一步都有详尽的指导。此外,还提供了测试安装是否成功的命令。

参考:https://github.com/NVIDIA/DIGITS/blob/digits-5.0/docs/InstallCuda.md#driver

#查看已安装驱动

cat /proc/driver/nvidia/version

NVRM version: NVIDIA UNIX x86_64 Kernel Module  384.90 

#卸载nvidia驱动

sudo apt-get remove --purge nvidia-384*

(用于ubuntu图形化界面循环登录,可先卸载nvidia驱动)

1,下载cuda驱动

http://www.nvidia.com/Download/index.aspx

2,安装 .deb文件

dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb
apt-get update
3,自动安装cuda toolkit和驱动

apt-get install cuda

或者:分别安装toolkit和drivers

  1. apt-get install cuda-toolkit-7-5 - Installs only the toolkit and not the driver.
  2. apt-get install cuda-drivers - Installs only the driver and not the toolkit.
4,配置环境变量

vi ~/.bashrc

##########添加内容###########

# CUDA
export LOCAL_DIR=/usr/local
export CUDA_75_HOME=$LOCAL_DIR/cuda-7.5
export CUDA_80_HOME=$LOCAL_DIR/cuda-8.0
export CUDA_HOME=$CUDA_80_HOME
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$CUDA_HOME/lib:$LD_LIBRARY_PATH


export CUDA_PATH=$CUDA_80_HOME

###########end########

source ~/.bashrc

5,重启

6,测试

nvcc -V

nvidia-smi

(使用.run安装cuda及nvidia驱动时不要选择装opengl)


Ubuntu系统上安装CUDA通常涉及几个主要步骤:确认硬件兼容性、准备环境以及通过合适的源或运行文件来完成安装。 ### 准备工作 1. **验证GPU支持** 需要先确保您的机器上有NVIDIA GPU,并且已正确识别。可以通过如下命令检查: ```bash lspci | grep -i nvidia ``` 2. **查看当前驱动版本和支持的CUDA工具集** 使用`nvidia-smi`可以显示有关显卡及其驱动的信息,同时也列出了该驱动所支持的最大CUDA Toolkit版本[^3]。 ```bash nvidia-smi ``` 3. **选择适合系统的CUDA版本** 不同的操作系统发行版可能推荐不同的CUDA版本。比如对于较新的Ubuntu 22.04 LTS来说,官方提供了针对CUDA 12.5的支持方案[^2];而如果是在旧一点的Ubuntu 18.04环境下,则更倾向于采用像CUDA 10.0这样的稳定长期维护版本[^1]。 ### 安装方法一:使用APT仓库(适用于最新Linux发行版) 此方法特别适合那些希望快速部署并且不需要特定历史版本用户的场景。 ```bash # 下载并配置存储库元数据 wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600 # 获取本地deb包并注册到dpkg数据库中 wget https://developer.download.nvidia.com/compute/cuda/12.5.0/local_installers/cuda-repo-ubuntu2204-12-5-local_12.5.0-555.42.02-1_amd64.deb sudo dpkg -i cuda-repo-ubuntu2204-12-5-local_12.5.0-555.42.02-1_amd64.deb # 导入GPG密钥环至指定目录 sudo cp /var/cuda-repo-ubuntu2204-12-5-local/cuda-*-keyring.gpg /usr/share/keyrings/ # 更新软件列表后开始正式安装所需组件 sudo apt-get update sudo apt-get -y install cuda-toolkit-12-5 ``` ### 安装方法二:手动下载.run文件进行自定义安装 这种方法允许更加灵活地控制整个流程,包括能够跳过某些不必要的部分如图形界面开发套件等。 ```bash # 替换URL为你想要的具体版本链接 wget https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux.run chmod +x cuda_*.run ./cuda_*.run --extract=`pwd` # 或者直接 ./cuda_*.run 开始交互式安装向导 ``` ### 后续操作 一旦基础架构搭建完毕,请记得调整PATH变量以便全局访问编译器及其他实用程序: ```bash export PATH=/usr/local/cuda/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda/lib64\ ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ``` 建议将这两行加入`.bashrc`或者相应的shell初始化脚本里以实现持久化设置。 最后测试一下是否成功激活了新环境: ```bash nvcc --version # 应显示出对应版本号 ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值