This is a project about reinforcement learning on Unity Agent and ROS. Here, I have provided some scenes for reference.
Component | Version | Remark |
---|---|---|
System | Ubuntu 22.04 | |
Unity Hub | 2022.3.6f1 | |
Python | 3.10.8 | 3.10.x |
Pytorch | 1.11.0+cu113 | For the CUDA |
ml-agents | Release 20 | Github Link |
CUDA | 11.3 | CUDA Link |
ROS2 | Humble |
You can follow this official doc or follow me to install in Ubuntu 22.04(Since you may have problem in installing here).
conda create -n ml-agment python=3.10.8
conda activate ml-agents
pip3 install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113
mkdir -p ~/Projects && cd ~/Projects
git clone --branch release_20 https://github.com/Unity-Technologies/ml-agents.git
cd ~/Projects
pip3 install -e ./ml-agents-envs
pip3 install -e ./ml-agents
You can follow the Offical Document.
sudo apt update && sudo apt upgrade -y
# or sudo apt update
ubuntu-drivers devices
sudo ubuntu-drivers autoinstall
# or sudo apt install nvidia-driver-535
reboot
nvidia-smi
wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda-repo-ubuntu2004-11-3-local_11.3.0-465.19.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-3-local_11.3.0-465.19.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-3-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda-11-3
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
tensorboard --logdir results