ubuntu22.04 openvino配置GPU
时间: 2024-12-20 17:31:04 浏览: 10
在Ubuntu 22.04上安装并配置OpenVINO支持GPU,你需要遵循几个步骤:
1. **更新系统**:
首先,确保你的系统是最新的,运行 `sudo apt update` 和 `sudo apt upgrade`.
2. **安装依赖**:
安装一些必要的软件包,包括CUDA、cuDNN和OpenCV,使用命令:
```bash
sudo apt install nvidia-cuda-toolkit nvidia-cudnn8-devel libopencv-dev
```
3. **添加驱动源**:
如果你的显卡不是默认支持的,可能需要添加额外的驱动源,比如使用NVIDIA的官方PPA:
```bash
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
```
4. **安装GPU驱动**:
根据你的GPU型号选择合适的驱动,例如对于NVIDIA GPU,输入 `nvidia-driver-[version]`, 替换 `[version]` 为你需要的具体版本。
5. **激活GPU**:
安装完驱动后,通过命令行激活GPU,并配置环境变量:
```bash
sudo nvidia-smi -q -d Memory |grep -A4 GPU|grep Free > tmp && echo "export CUDA_VISIBLE_DEVICES=all" >> ~/.bashrc && echo "echo \$?" >> ~/.bashrc && source ~/.bashrc
```
6. **安装OpenVINO**:
使用官方的安装脚本,访问 https://docs.openvinotoolkit.org/latest/openvino_installation_downloads_ubuntu.html 下载适用于GPU的版本,然后按照指示安装。
7. **验证配置**:
安装完成后,你可以通过运行 `openvino_info` 或者 `python -c "import openvino"` 来确认OpenVINO是否已经成功识别到GPU。
阅读全文