Ubuntu20.04开发程序调用GPU
时间: 2024-10-27 18:13:40 浏览: 14
Ubuntu 20.04 开发程序调用 GPU 的步骤通常涉及安装必要的库和驱动,以及设置环境变量。以下是基本的流程:
1. **更新系统**:
```
sudo apt update && sudo apt upgrade
```
2. **安装CUDA和cuDNN** (如果你需要深度学习相关的GPU计算):
- 添加CUDA和NVIDIA包源:
```bash
sudo add-apt-repository ppa:ubuntu-toolchain-r/test
sudo add-apt-repository ppa:nvidia CORPORATION/nvidia-450
- 安装CUDA:
```bash
sudo apt install nvidia-driver-450 cuda-cudart-dev
```
- 如果需要cuDNN:
```bash
wget https://developer.nvidia.com/embedded/cuda-zip-archive -O cuda-repo-ubuntu2004-11-4-ga-ubuntu20.04_11.4.3-1_amd64.deb
sudo dpkg -i cuda-repo-*.deb
sudo apt update
sudo apt install libcudnn7 libcudnn7-dev
```
3. **安装其他GPU支持库** (如OpenGL、GLUT等):
```bash
sudo apt-get install libgl1-mesa-dev freeglut3-dev mesa-utils
```
4. **设置环境变量**:
- 创建一个名为 `.bashrc` 或者 `.zshrc` 的文件,添加:
```bash
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export CUDACXX=/usr/local/cuda/bin/nvcc
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CUDA_HOME/lib64
```
5. **测试**:
你可以运行一些简单的示例程序来确认是否成功连接到GPU,例如 `nvidia-smi` 查看GPU状态。
阅读全文