https://yuansima.com/tags/rocm/ Recent content in ROCm on 远思码技术实验室 Hugo -- gohugo.io zh-cn Sun, 15 Feb 2026 00:00:00 +0000 https://yuansima.com/p/pytorch-rocm-setup/ Sun, 15 Feb 2026 00:00:00 +0000 https://yuansima.com/p/pytorch-rocm-setup/ <p>众所周知，深度学习领域长期被 CUDA 统治。但随着 AMD ROCm 平台的成熟，以及性价比的诱惑，越来越多的炼丹师开始尝试“红队”方案。</p> <p>作为远思技术实验室的基础设施之一，我在配置 <strong>PyTorch + ROCm</strong> 环境时遇到了不少坑，特此记录，希望能帮到后来人。</p> <h2 id="为什么选择-docker">为什么选择 Docker？ </h2><p>直接在宿主机（Host）上安装 ROCm 驱动和编译器极易导致依赖地狱（Dependency Hell）。使用 Docker 可以将环境隔离，随时回滚，是最佳实践。</p> <h2 id="核心步骤">核心步骤 </h2><h3 id="1-基础镜像选择">1. 基础镜像选择 </h3><p>不要尝试自己从 scratch 构建，直接拉取 AMD 官方提供的 PyTorch 镜像：</p> <div class="highlight"><pre tabindex="0" style="color:#f8f8f2;background-color:#272822;-moz-tab-size:4;-o-tab-size:4;tab-size:4;"><code class="language-bash" data-lang="bash"><span style="display:flex;"><span>docker pull rocm/pytorch:latest </span></span></code></pre></div>