IQ-Learn 在 RTX 3090 服务器上的环境配置与踩坑记录

news2026/5/5 6:00:49
最近在远程服务器上配置 IQ-Learn 的强化学习环境时按照项目的 requirements.txt 直接安装依赖过程中连续遇到了多个老项目兼容性问题。这里把排查和修复过程整理下来方便以后快速复现。1.项目依赖项目的 requirements.txt 如下gym[box2d]0.17.1hydra-core1.0.6stable_baselines31.0tensorboard2.4.0tensorboardX2.1torch1.7.1torchvision0.8.2tornado5.1.1tqdm4.42.1wandbopencv-python4.5.1.48atari-py0.2.6gym_minigrid1.0.2mujoco_py2.0.2项目 README 中说明pytorch 1.4hydra-core1.0hydra-core 1.1 当前不兼容因此真正不能随便升级的是 hydra-core而 torch 虽然写得较老但实际上可以根据 GPU 做适配。2. 原始环境的问题一开始直接执行pip install -r requirements.txt报错主要集中在以下几个方面2.1 mujoco-py 找不到动态库路径报错提示缺少LD_LIBRARY_PATH说明 MuJoCo 路径没有正确导出。2.2 box2d-py 编译失败报错error: command swig failed: No such file or directory说明服务器缺少 swig。2.3 mujoco-py 与 Cython 3.x 不兼容报错涉及 noexcept / Cython.Compiler.Errors.CompileError本质是mujoco-py2.0.2.0不兼容 Cython 3.x2.4 stable_baselines3 / tensorboard / protobuf 冲突报错中出现Descriptors cannot be created directly这是旧 tensorboard 与新 protobuf 不兼容。2.5 tensorboard / stable_baselines3 与 numpy 冲突报错AttributeError: module numpy has no attribute object这是因为旧版 tensorboard 还在使用 np.object。2.6 wandb 版本过新运行 train_iq.py 时wandb 对 Hydra/OmegaConf 配置对象的处理不兼容报错TypeError: first argument must be callable or None2.7 torch 1.7.1 无法驱动 RTX 3090运行训练时出现NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation说明 torch1.7.1 太老不支持 sm_86 架构。3. 最终思路单独创建适配 3090 的新环境为了兼容 RTX 3090我没有继续使用原来的 torch1.7.1而是新建了一个环境 irl核心策略是保留项目真正敏感的老依赖hydra-core1.0.6stable_baselines31.0tensorboard2.4.0将 torch / torchvision 升级到支持 3090 的版本torch1.10.2cu113torchvision0.11.3cu113对其他依赖做兼容性回退Cython0.29.37protobuf3.20.3numpy1.23.5wandb0.10.33setuptools59.5.04. 最终安装命令4.1 创建环境conda create -n irl python3.8 -y conda activate irl4.2 安装兼容旧 Hydra/OmegaConf 的 pip 和 setuptoolspip install --no-cache-dir pip24.1 pip install --no-cache-dir setuptools59.5.0说明pip24.1 会拒绝 omegaconf 2.0.x 的旧元数据格式较新的 setuptools 会让旧版 tensorboard/torch.utils.tensorboard 出现 distutils 兼容问题4.3 安装编译工具conda install -n irl -c conda-forge swig cmake4 make -y说明swig 用于 box2d-pycmake4 用于兼容 atari-py0.2.64.4 安装支持 RTX 3090 的 PyTorchpip install --no-cache-dir \ torch1.10.2cu113 \ torchvision0.11.3cu113 \ --extra-index-url https://download.pytorch.org/whl/cu1134.5 安装主依赖pip install --no-cache-dir \ gym[box2d]0.17.1 \ hydra-core1.0.6 \ stable_baselines31.0 \ tensorboard2.4.0 \ tensorboardX2.1 \ tornado5.1.1 \ tqdm4.42.1 \ opencv-python4.5.1.48 \ gym_minigrid1.0.2 \ termcolor4.6 安装兼容性固定版本pip install --no-cache-dir \ Cython0.29.37 \ numpy1.23.5 \ protobuf3.20.3 \ wandb0.10.334.7 配置 MuJoCo 路径export MUJOCO_PY_MUJOCO_PATH/home/duweicheng/.mujoco/mujoco200 export LD_LIBRARY_PATH$LD_LIBRARY_PATH:/home/duweicheng/.mujoco/mujoco200/bin为了避免每次登录都重新设置可以写入 ~/.bashrcecho export MUJOCO_PY_MUJOCO_PATH/home/duweicheng/.mujoco/mujoco200 ~/.bashrc echo export LD_LIBRARY_PATH$LD_LIBRARY_PATH:/home/duweicheng/.mujoco/mujoco200/bin ~/.bashrc source ~/.bashrc4.8 安装 Box2D、MuJoCo、Atari 相关依赖先安装 box2d-pypip install --no-cache-dir box2d-py2.3.8再安装 mujoco-py 的构建依赖pip install --no-cache-dir \ cffi1.17.1 \ glfw2.10.0 \ imageio2.35.1 \ lockfile0.12.2安装 mujoco-py 时关闭构建隔离pip install --no-cache-dir --no-build-isolation mujoco-py2.0.2.0最后安装 Ataripip install --no-cache-dir atari-py0.2.65. 如何验证安装成功5.1 验证 PyTorch 和 3090 是否可用python -c import torch; print(torch:, torch.__version__); print(cuda available:, torch.cuda.is_available()); print(device 0:, torch.cuda.get_device_name(0))以及做一次真实的 CUDA 计算python -c import torch; xtorch.randn(1024,1024, devicecuda); ytorch.randn(1024,1024, devicecuda); zxy; print(cuda matmul ok:, z.shape)如果这两步通过说明PyTorch 版本支持 3090CUDA 驱动和当前 wheel 匹配GPU 计算是可用的5.2 验证主要依赖导入成功python -c from stable_baselines3 import PPO; print(sb3 ok) python -c from torch.utils.tensorboard import SummaryWriter; print(tensorboard ok) python -c import mujoco_py; print(mujoco_py ok) python -c import atari_py; print(atari_py ok)5.3 运行训练命令CUDA_VISIBLE_DEVICES1 python train_iq.py envhopper agentsac expert.demos1 method.lossv0 method.regularizeTrue agent.actor_lr3e-5 seed0如果能进入训练流程就说明整个环境已经可用。6. 这次实际解决了什么问题本次环境配置最终解决了以下问题mujoco-py 缺少 LD_LIBRARY_PATHbox2d-py 需要 swigmujoco-py 与 Cython 3.x 不兼容tensorboard 2.4.0 与 protobuf 5.x 不兼容tensorboard 2.4.0 与 numpy 1.24 不兼容过新 wandb 与旧版 Hydra/OmegaConf 配置对象不兼容torch 1.7.1 不支持 RTX 3090 (sm_86)新版 pip 不兼容旧 omegaconf 2.0.x新版 setuptools 导致旧版 torch.utils.tensorboard 的 distutils 兼容问题atari-py 与 cmake 4.x 不兼容项目还缺少 termcolor 这个未写入 requirements.txt 的依赖7. 最终可用环境版本下面是最终验证通过的核心版本组合python3.8 pip24.0 setuptools59.5.0 torch1.10.2cu113 torchvision0.11.3cu113 gym0.17.1 hydra-core1.0.6 stable_baselines31.0 tensorboard2.4.0 tensorboardX2.1 tornado5.1.1 tqdm4.42.1 opencv-python4.5.1.48 gym_minigrid1.0.2 atari-py0.2.6 mujoco-py2.0.2.0 box2d-py2.3.8 Cython0.29.37 numpy1.23.5 protobuf3.20.3 wandb0.10.33 setuptools59.5.0 termcolor*8. 最终环境打印结果安装并验证通过后环境关键输出如下torch: 1.10.2cu113 cuda available: True device 0: NVIDIA GeForce RTX 3090 cuda matmul ok: torch.Size([1024, 1024]) sb3 ok tensorboard ok mujoco_py ok atari_py ok这说明GPU 可用3090 可被当前 PyTorch 正常驱动stable_baselines3 可导入tensorboard 可导入mujoco_py 可导入atari_py 可导入环境已经满足 IQ-Learn 在 MuJoCo、Atari 等任务上的运行需求。9. 整理好的完整bash指令#!/usr/bin/env bashset -euo pipefailENV_NAMEirlPYTHON_VERSION3.8MUJOCO_PATH/home/duweicheng/.mujoco/mujoco200echo Creating conda environment: ${ENV_NAME}conda create -n ${ENV_NAME} python${PYTHON_VERSION} -yecho Activating environment: ${ENV_NAME}source $(conda info --base)/etc/profile.d/conda.shconda activate ${ENV_NAME}echo Downgrading pip for old hydra/omegaconf compatibilitypip install --no-cache-dir pip24.1pip install --no-cache-dir setuptools59.5.0echo Installing build toolsconda install -n ${ENV_NAME} -c conda-forge swig cmake4 make -yecho Installing PyTorch for RTX 3090pip install --no-cache-dir \torch1.10.2cu113 \torchvision0.11.3cu113 \--extra-index-url https://download.pytorch.org/whl/cu113echo Installing base project dependenciespip install --no-cache-dir \gym[box2d]0.17.1 \hydra-core1.0.6 \stable_baselines31.0 \tensorboard2.4.0 \tensorboardX2.1 \tornado5.1.1 \tqdm4.42.1 \opencv-python4.5.1.48 \gym_minigrid1.0.2 \termcolorecho Installing compatibility-pinned packagespip install --no-cache-dir \Cython0.29.37 \numpy1.23.5 \protobuf3.20.3 \wandb0.10.33echo Exporting MuJoCo environment variables for current sessionexport MUJOCO_PY_MUJOCO_PATH${MUJOCO_PATH}export LD_LIBRARY_PATH${LD_LIBRARY_PATH:-}:${MUJOCO_PATH}/binecho Checking MuJoCo pathtest -d ${MUJOCO_PATH}/binecho Installing Box2Dpip install --no-cache-dir box2d-py2.3.8echo Installing explicit MuJoCo build depspip install --no-cache-dir \cffi1.17.1 \glfw2.10.0 \imageio2.35.1 \lockfile0.12.2echo Installing mujoco-py without build isolationpip install --no-cache-dir --no-build-isolation mujoco-py2.0.2.0echo Installing Atari dependencycmake --versionpip install --no-cache-dir atari-py0.2.6echo Persisting MuJoCo environment variables to ~/.bashrc if missinggrep -qxF export MUJOCO_PY_MUJOCO_PATH${MUJOCO_PATH} ~/.bashrc || \echo export MUJOCO_PY_MUJOCO_PATH${MUJOCO_PATH} ~/.bashrcgrep -qxF export LD_LIBRARY_PATH\$LD_LIBRARY_PATH:${MUJOCO_PATH}/bin ~/.bashrc || \echo export LD_LIBRARY_PATH\$LD_LIBRARY_PATH:${MUJOCO_PATH}/bin ~/.bashrcecho Verifying installationpython -c import torch; print(torch:, torch.__version__); print(cuda available:, torch.cuda.is_available()); print(device 0:, torch.cuda.get_device_name(0))python -c import torch; xtorch.randn(1024,1024, devicecuda); ytorch.randn(1024,1024, devicecuda); zxy; print(cuda matmul ok:, z.shape)python -c from stable_baselines3 import PPO; print(sb3 ok)python -c from torch.utils.tensorboard import SummaryWriter; print(tensorboard ok)python -c import mujoco_py; print(mujoco_py ok)python -c import atari_py; print(atari_py ok)echo Doneecho Activate with: conda activate ${ENV_NAME}echo Run Hopper training with:echo CUDA_VISIBLE_DEVICES1 python train_iq.py envhopper agentsac expert.demos1 method.lossv0 method.regularizeTrue agent.actor_lr3e-5 seed010.现有的环境包Package Version---------------------- ------------absl-py 2.3.1antlr4-python3-runtime 4.8atari-py 0.2.6box2d-py 2.3.8cachetools 4.2.4certifi 2026.4.22cffi 1.17.1charset-normalizer 3.4.7click 8.1.8cloudpickle 1.3.0configparser 7.1.0contourpy 1.1.1cycler 0.12.1Cython 0.29.37docker-pycreds 0.4.0fonttools 4.57.0future 1.0.0gitdb 4.0.12GitPython 3.1.49glfw 2.10.0google-auth 1.35.0google-auth-oauthlib 0.4.6grpcio 1.70.0gym 0.17.1gym-minigrid 1.0.2hydra-core 1.0.6idna 3.13imageio 2.35.1importlib_metadata 8.5.0importlib_resources 6.4.5kiwisolver 1.4.7lockfile 0.12.2Markdown 3.7MarkupSafe 2.1.5matplotlib 3.7.5mujoco-py 2.0.2.0numpy 1.23.5nvidia-ml-py 13.590.48nvitop 1.6.2oauthlib 3.3.1omegaconf 2.0.6opencv-python 4.5.1.48packaging 26.2pandas 2.0.3pathtools 0.1.2pillow 10.4.0pip 24.0promise 2.3protobuf 3.20.3psutil 7.2.2pyasn1 0.6.3pyasn1_modules 0.4.2pycparser 2.23pyglet 1.5.0pyparsing 3.1.4python-dateutil 2.9.0.post0pytz 2026.2PyYAML 6.0.3requests 2.32.4requests-oauthlib 2.0.0rsa 4.9.1scipy 1.10.1sentry-sdk 2.58.0setuptools 59.5.0shortuuid 1.0.13six 1.17.0smmap 5.0.3stable-baselines3 1.0subprocess32 3.5.4tensorboard 2.4.0tensorboard-plugin-wit 1.8.1tensorboardX 2.1termcolor 2.4.0torch 1.10.2cu113torchvision 0.11.3cu113tornado 5.1.1tqdm 4.42.1typing_extensions 4.13.2tzdata 2026.2urllib3 2.2.3wandb 0.10.33Werkzeug 3.0.6wheel 0.44.0zipp 3.20.2hrefblob:https://mp.csdn.net/9b283dfd-623b-4845-afc2-e02622ff2c66 relstylesheet /

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2584072.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

SpringBoot-17-MyBatis动态SQL标签之常用标签

文章目录 1 代码1.1 实体User.java1.2 接口UserMapper.java1.3 映射UserMapper.xml1.3.1 标签if1.3.2 标签if和where1.3.3 标签choose和when和otherwise1.4 UserController.java2 常用动态SQL标签2.1 标签set2.1.1 UserMapper.java2.1.2 UserMapper.xml2.1.3 UserController.ja…

wordpress后台更新后 前端没变化的解决方法

使用siteground主机的wordpress网站,会出现更新了网站内容和修改了php模板文件、js文件、css文件、图片文件后,网站没有变化的情况。 不熟悉siteground主机的新手,遇到这个问题,就很抓狂,明明是哪都没操作错误&#x…

网络编程(Modbus进阶)

思维导图 Modbus RTU(先学一点理论) 概念 Modbus RTU 是工业自动化领域 最广泛应用的串行通信协议,由 Modicon 公司(现施耐德电气)于 1979 年推出。它以 高效率、强健性、易实现的特点成为工业控制系统的通信标准。 包…

UE5 学习系列(二)用户操作界面及介绍

这篇博客是 UE5 学习系列博客的第二篇,在第一篇的基础上展开这篇内容。博客参考的 B 站视频资料和第一篇的链接如下: 【Note】:如果你已经完成安装等操作,可以只执行第一篇博客中 2. 新建一个空白游戏项目 章节操作,重…

IDEA运行Tomcat出现乱码问题解决汇总

最近正值期末周,有很多同学在写期末Java web作业时,运行tomcat出现乱码问题,经过多次解决与研究,我做了如下整理: 原因: IDEA本身编码与tomcat的编码与Windows编码不同导致,Windows 系统控制台…

利用最小二乘法找圆心和半径

#include <iostream> #include <vector> #include <cmath> #include <Eigen/Dense> // 需安装Eigen库用于矩阵运算 // 定义点结构 struct Point { double x, y; Point(double x_, double y_) : x(x_), y(y_) {} }; // 最小二乘法求圆心和半径 …

使用docker在3台服务器上搭建基于redis 6.x的一主两从三台均是哨兵模式

一、环境及版本说明 如果服务器已经安装了docker,则忽略此步骤,如果没有安装,则可以按照一下方式安装: 1. 在线安装(有互联网环境): 请看我这篇文章 传送阵>> 点我查看 2. 离线安装(内网环境):请看我这篇文章 传送阵>> 点我查看 说明&#xff1a;假设每台服务器已…

XML Group端口详解

在XML数据映射过程中&#xff0c;经常需要对数据进行分组聚合操作。例如&#xff0c;当处理包含多个物料明细的XML文件时&#xff0c;可能需要将相同物料号的明细归为一组&#xff0c;或对相同物料号的数量进行求和计算。传统实现方式通常需要编写脚本代码&#xff0c;增加了开…

LBE-LEX系列工业语音播放器|预警播报器|喇叭蜂鸣器的上位机配置操作说明

LBE-LEX系列工业语音播放器|预警播报器|喇叭蜂鸣器专为工业环境精心打造&#xff0c;完美适配AGV和无人叉车。同时&#xff0c;集成以太网与语音合成技术&#xff0c;为各类高级系统&#xff08;如MES、调度系统、库位管理、立库等&#xff09;提供高效便捷的语音交互体验。 L…

(LeetCode 每日一题) 3442. 奇偶频次间的最大差值 I (哈希、字符串)

题目&#xff1a;3442. 奇偶频次间的最大差值 I 思路 &#xff1a;哈希&#xff0c;时间复杂度0(n)。 用哈希表来记录每个字符串中字符的分布情况&#xff0c;哈希表这里用数组即可实现。 C版本&#xff1a; class Solution { public:int maxDifference(string s) {int a[26]…

【大模型RAG】拍照搜题技术架构速览:三层管道、两级检索、兜底大模型

摘要 拍照搜题系统采用“三层管道&#xff08;多模态 OCR → 语义检索 → 答案渲染&#xff09;、两级检索&#xff08;倒排 BM25 向量 HNSW&#xff09;并以大语言模型兜底”的整体框架&#xff1a; 多模态 OCR 层 将题目图片经过超分、去噪、倾斜校正后&#xff0c;分别用…

【Axure高保真原型】引导弹窗

今天和大家中分享引导弹窗的原型模板&#xff0c;载入页面后&#xff0c;会显示引导弹窗&#xff0c;适用于引导用户使用页面&#xff0c;点击完成后&#xff0c;会显示下一个引导弹窗&#xff0c;直至最后一个引导弹窗完成后进入首页。具体效果可以点击下方视频观看或打开下方…

接口测试中缓存处理策略

在接口测试中&#xff0c;缓存处理策略是一个关键环节&#xff0c;直接影响测试结果的准确性和可靠性。合理的缓存处理策略能够确保测试环境的一致性&#xff0c;避免因缓存数据导致的测试偏差。以下是接口测试中常见的缓存处理策略及其详细说明&#xff1a; 一、缓存处理的核…

龙虎榜——20250610

上证指数放量收阴线&#xff0c;个股多数下跌&#xff0c;盘中受消息影响大幅波动。 深证指数放量收阴线形成顶分型&#xff0c;指数短线有调整的需求&#xff0c;大概需要一两天。 2025年6月10日龙虎榜行业方向分析 1. 金融科技 代表标的&#xff1a;御银股份、雄帝科技 驱动…

观成科技:隐蔽隧道工具Ligolo-ng加密流量分析

1.工具介绍 Ligolo-ng是一款由go编写的高效隧道工具&#xff0c;该工具基于TUN接口实现其功能&#xff0c;利用反向TCP/TLS连接建立一条隐蔽的通信信道&#xff0c;支持使用Let’s Encrypt自动生成证书。Ligolo-ng的通信隐蔽性体现在其支持多种连接方式&#xff0c;适应复杂网…

铭豹扩展坞 USB转网口 突然无法识别解决方法

当 USB 转网口扩展坞在一台笔记本上无法识别,但在其他电脑上正常工作时,问题通常出在笔记本自身或其与扩展坞的兼容性上。以下是系统化的定位思路和排查步骤,帮助你快速找到故障原因: 背景: 一个M-pard(铭豹)扩展坞的网卡突然无法识别了,扩展出来的三个USB接口正常。…

未来机器人的大脑:如何用神经网络模拟器实现更智能的决策?

编辑&#xff1a;陈萍萍的公主一点人工一点智能 未来机器人的大脑&#xff1a;如何用神经网络模拟器实现更智能的决策&#xff1f;RWM通过双自回归机制有效解决了复合误差、部分可观测性和随机动力学等关键挑战&#xff0c;在不依赖领域特定归纳偏见的条件下实现了卓越的预测准…

Linux应用开发之网络套接字编程(实例篇)

服务端与客户端单连接 服务端代码 #include <sys/socket.h> #include <sys/types.h> #include <netinet/in.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <arpa/inet.h> #include <pthread.h> …

华为云AI开发平台ModelArts

华为云ModelArts&#xff1a;重塑AI开发流程的“智能引擎”与“创新加速器”&#xff01; 在人工智能浪潮席卷全球的2025年&#xff0c;企业拥抱AI的意愿空前高涨&#xff0c;但技术门槛高、流程复杂、资源投入巨大的现实&#xff0c;却让许多创新构想止步于实验室。数据科学家…

深度学习在微纳光子学中的应用

深度学习在微纳光子学中的主要应用方向 深度学习与微纳光子学的结合主要集中在以下几个方向&#xff1a; 逆向设计 通过神经网络快速预测微纳结构的光学响应&#xff0c;替代传统耗时的数值模拟方法。例如设计超表面、光子晶体等结构。 特征提取与优化 从复杂的光学数据中自…