今天使用
fanziqi
大佬的rl_docker搭建了一个isaac gym
下的四足机器人训练环境,成功运行legged gym
项目下的例子,记录一下搭建流程
ubuntu下可使用
hostnamectl
查看配置。
使用 fishros
大佬的一键安装脚本,里面有对于的自动识别并更换最快的国内系统源
使用命令:
wget http://fishros.com/install -O fishros && . fishros
然后安装出现的选项选择更换系统源
即可。
nvidia-smi
!![[Pasted image 20241206232824.png|800]]
我这里使用 Driver Version: 550.120 CUDA Version: 12.4
可以成功运行 isaac gym
的例程。
注意,我一开始是参考csdn一篇文章手动安装的nvidia驱动,后面运行的时候会出现问题。解决方法看后面 尝试了降低版本依然不可以,出现了以下错误:
输入命令:nvidia-smi
出现:
Failed to initialize NVML: Driver/library version mismatch NVML library version: 550.120
在网上查找了很多方法,总结如下:(虽然没解决问题, 但是学习了其他情况下的解决办法和一些linux下的指令)
nvidia-smi
打不开的情况 cat /proc/driver/nvidia/version
![[Pasted image 20241206234151.png|800]]
dpkg -l | grep nvidia-driver
![[Pasted image 20241206234449.png|800]]
如果驱动和内核版本不对应,只需要将驱动升级到和内核一样的版本即可。
参考# 如何在Ubuntu系统中彻底卸载和删除NVIDIA显卡驱动程序 和 # nvidia-smi is returning “NVML: Driver/library version mismatch”.
sudo apt-get --purge remove "*nvidia*"
sudo apt-get --purge remove "*libnvidia*"
sudo apt-get --purge remove cuda
sudo apt autoremove
如果使用了手动安装驱动,即./NVIDIA-xxx-.run
则使用:
sudo NVIDIA-Linux-x86_64-<version>.run --uninstall
将
替换为你的驱动版本号。
参考链接:# 如何在Ubuntu系统中彻底卸载和删除NVIDIA显卡驱动程序
sudo systemctl stop gdm
sudo rmmod nvidia_uvm
sudo rmmod nvidia_drm
sudo rmmod nvidia_modeset
sudo rmmod nvidia
rl_docker rsl-rl legged_gym isaacgym
rl_docker
修改docker/Dockerfile
文件中的第一句为:
nvcr.io/nvidia/pytorch:22.12-py3
在 run.sh
中加入 chmod +x ${SCRIPTROOT}/…/rl_docker/setup.sh
在 Dockerfile
中如果:
RUN apt-get install -q -y --no-install-recommends wget \
&& wget http://archive.ubuntu.com/ubuntu/pool/main/c/ca-certificates/ca-certificates_20240203~20.04.1_all.deb \
&& dpkg -i ./ca-certificates_20240203~20.04.1_all.deb
出现错误,在官网查找最新的连接即可。
rl_docker
中的bash build.sh
之前,先安装 rsl-rl legged_gym isaacgym
.
rsl_rl
要安装旧版本 Pre-Release v1.0.2legged_gym
安装isaacgym
安装rl_docker
中步骤进行即可isaacgym
程序:rl_ws
下执行 cd ./isaacgym/python/examples/ && python ./joint_monkey.py
isaacgym
.legged_gym
程序rl_ws
下执行 cd ./legged_gym/ && python legged_gym/scripts/train.py --task=anymal_c_flat --num_envs=1024
参考连接:# 强化学习仿真器Isaac Gym的安装与配置