ubuntu20.04 ROCM
时间: 2023-10-17 10:06:24 浏览: 242
在Ubuntu 20.04上安装ROCM的步骤如下:
1. 添加ROCm apt存储库:使用以下命令安装wget和gnupg2:
```
sudo apt install wget gnupg2
```
然后使用wget命令下载ROCM的GPG密钥,并将其添加到apt密钥环中:
```
wget -q -O - https://repo.radeon.com/rocm/rocm.gpg.key | sudo apt-key add -
```
接下来,使用以下命令将ROCM的apt存储库添加到/etc/apt/sources.list.d/rocm.list文件中:
```
echo 'deb [arch=amd64 https://repo.radeon.com/rocm/apt/4.3/ ubuntu main' | sudo tee /etc/apt/sources.list.d/rocm.list
```
2. 安装ROCm元软件包:首先更新存储库列表,然后安装rocm-dkms元软件包并重新启动系统:
```
sudo apt update
sudo apt install rocm-dkms && sudo reboot
```
3. 更新系统:运行以下命令以确保系统是最新的:
```
sudo apt update
sudo apt dist-upgrade
sudo apt install libnuma-dev
sudo reboot
```
这些步骤将帮助您在Ubuntu 20.04上安装ROCM。请注意,您需要具有sudo权限才能执行这些命令。
相关问题
ubuntu20.04安装vllm
### 安装 VLLM 的前提条件
为了在 Ubuntu 20.04 上成功安装并配置 VLLM,需先确保系统已准备好必要的依赖项。这包括但不限于更新软件包列表以及安装 NVIDIA 驱动程序和相应的工具集。完成这些操作之后,通常建议重启系统以便新安装的驱动能够正常工作[^3]。
### 获取 vLLM ROCm Docker 镜像
接下来,在构建好适用于 vLLM 的 ROCm 版本 Docker 镜像后,可以通过特定命令来启动该镜像对应的容器实例。如果计划在容器内部署具体的模型,则应指定 `-v <path/to/model>:/app/models` 参数以实现主机目录向容器内的映射;反之,当不存在待挂载的模型时可省略此部分设置[^2]。
### 运行 vLLM 应用
对于希望进一步了解如何调用底层引擎执行具体任务的情况而言,涉及到了 `LLM._run_engine()` 方法及其后续流程中的 `_run_workers` 函数调用链路。这一过程负责协调多个工作者线程共同完成复杂的计算作业[^1]。
综上所述,以下是针对 Ubuntu 20.04 用户准备环境并部署 vLLM 所必需遵循的一系列指导:
#### 更新与重启
```bash
sudo apt update && sudo apt upgrade -y
sudo reboot
```
#### 安装Docker及相关组件(假设已经按照官方文档完成了ROCM平台搭建)
```bash
curl -fsSL https://get.docker.com | sh
sudo usermod -aG docker $USER
newgrp docker
```
#### 下载并运行vLLM-ROCm镜像
```bash
docker pull vllm-rocm
docker run -it --network=host \
--group-add video --ipc=host \
--cap-add SYS_PTRACE --security-opt seccomp=unconfined \
--device /dev/kfd --device /dev/dri \
-v </path/to/your/local/model/folder>:/app/models \
vllm-rocm
```
请注意上述脚本中 `/path/to/your/local/model/folder` 处应当替换为实际存在的本地路径名称。
ubuntu20.04安装amd驱动、
1. 打开终端,更新系统
```
sudo apt update
sudo apt upgrade
```
2. 安装AMDGPU-PRO驱动
```
wget https://drivers.amd.com/drivers/linux/amdgpu-pro-20.10-1048554-ubuntu-20.04.tar.xz
tar -Jxvf amdgpu-pro-20.10-1048554-ubuntu-20.04.tar.xz
cd amdgpu-pro-20.10-1048554-ubuntu-20.04
sudo ./amdgpu-install --opencl=legacy,pal --headless
```
3. 重启系统
```
sudo reboot
```
4. 验证驱动是否安装成功
```
sudo lshw -c video
```
如果成功安装,输出信息中应该会出现 "configuration: driver=amdgpu"。
5. 安装AMD GPU开发工具包(可选)
```
sudo apt install rocm-dev
```
阅读全文
相关推荐















