再有GPU下裝SurRoL
先裝

conda create -n surrol python=3.7 -y
conda activate surrol

 pip install gym==0.15.6
sudo apt-get install libopenmpi-dev zlib1g-dev
pip install tensorflow-gpu==1.14

進入python環境後,確認TensorFlow是否可成功連結至GPU

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

git clone https://github.com/med-air/SurRoL.git
cd SurRoL

裝openai baselines

sudo snap install cmake --classic
git clone https://github.com/openai/baselines
cd baselines
pip install -e .
pip install filelock  matplotlib pandas gym[atari]
conda install -c conda-forge ffmpeg==4.2.2

#測試baselines
pip install pytest
pytest

裝mpi4py

sudo apt install libopenmpi-dev
pip install mpi4py

裝SurRoL

cd ~/SurRoL
pip install -e .