当前位置：首页 > 系统教程 > 正文

Ubuntu 20.04 强化学习环境搭建全攻略（从零开始配置 PyTorch 与 Gymnasium）

主机测评网
系统教程
2026-04-02
950

Ubuntu 20.04 强化学习环境搭建全攻略（从零开始配置 PyTorch 与 Gymnasium）

SEO关键词：Ubuntu强化学习环境, CUDA安装教程, PyTorch配置, Gymnasium安装

一、下载 Ubuntu 20.04 ISO 镜像

首先，访问 Ubuntu 官网或国内镜像站（如清华大学开源软件镜像站）下载 Ubuntu 20.04.6 LTS 桌面版 ISO 镜像。选择 LTS 版本是因为其稳定性高，且主流深度学习框架对其支持最完善。

二、制作系统启动盘

准备一个 8GB 以上的 U 盘，使用 Rufus（Windows）或 BalenaEtcher（多平台）工具。将下载好的 ISO 镜像写入 U 盘。注意：此操作会清空 U 盘数据，请提前备份。

Ubuntu 20.04 强化学习环境搭建全攻略（从零开始配置 PyTorch 与 Gymnasium） Ubuntu强化学习环境 CUDA安装教程 PyTorch配置 Gymnasium安装第1张

三、安装 Ubuntu 系统

将 U 盘插入电脑，开机进入 BIOS 设置 U 盘启动。安装过程中建议：

选择 "Minimal Installation"（最小安装）以减少冗余软件。
勾选 "Install third-party software" 以获取更好的硬件支持。
手动分区时，建议为 / 根目录和 /home 目录预留充足空间。

四、更换 pip 与系统软件源

为了提高下载速度，我们需要更换国内镜像源。打开终端执行：

# 更换 pip 源pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple# 更新系统软件源（手动在"Software & Updates"中选择阿里云或清华源）sudo apt update && sudo apt upgrade -y

五、安装 NVIDIA 驱动、CUDA 与 cuDNN

这是配置 Ubuntu强化学习环境 的核心步骤：

驱动：在"Additional Drivers"中选择带有 (recommended) 的 NVIDIA 驱动并应用。
CUDA：前往 NVIDIA 官网下载 CUDA 11.8（推荐版本），执行 sudo sh cuda_xxx.run。
cuDNN：下载对应的 cuDNN 压缩包，解压并将 include 和 lib64 文件夹下的内容复制到 CUDA 安装路径中。

六、安装 PyTorch 与 Gymnasium

在完成 CUDA安装教程 后，接下来配置深度学习与强化学习环境：

# 安装 PyTorch (根据CUDA版本选择)pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118# 安装强化学习必备库 Gymnasiumpip install gymnasium# 安装包含所有物理引擎环境的扩展包\pip install gymnasium[all]

七、环境测试

最后，运行以下 Python 代码验证 PyTorch配置 与 Gymnasium安装 是否成功：

import torchimport gymnasium as gymprint(f'CUDA 可用性: {torch.cuda.is_available()}')env = gym.make('CartPole-v1', render_mode='human')observation, info = env.reset()print('环境启动成功！')

至此，您的强化学习工作站已搭建完毕！