ubuntu安装cuda示例包
时间: 2024-10-06 21:00:25 浏览: 45
Linux Ubuntu 安装 PyTorch 示例
Ubuntu上安装CUDA示例包通常涉及几个步骤,因为CUDA是NVIDIA专有的GPU加速计算平台,需要先安装NVIDIA驱动并配置好环境。以下是简要步骤:
1. **更新系统**:
```bash
sudo apt-get update
sudo apt-get upgrade
```
2. **添加NVIDIA官方存储库**:
```bash
sudo sh -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ $(lsb_release -cs) main" > /etc/apt/sources.list.d/nvidia.list'
```
对于其他版本的Ubuntu,替换相应的代号。
3. **安装密钥**:
```bash
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/gpgkey
```
4. **安装CUDA**:
使用`apt`安装CUDA,例如CUDA 11.x:
```bash
sudo apt-get install nvidia-driver-cuda-11-0 nvidia-container-toolkit
```
或者安装特定版本的CUDA和相关工具:
```bash
sudo apt-get install cuda-11-0 cuDNN-11-0
```
5. **安装示例程序**:
CUDA提供了一些示例代码,可以从中学习CUDA编程。运行下面的命令来安装它们:
```bash
cd /usr/local/cuda-examples
sudo make -j$(nproc)
```
6. **验证安装**:
查看CUDA是否成功安装,你可以运行一些CUDA提供的示例程序如 `cuobjdump` 或 `nvidia-smi`。
阅读全文