40 lines
864 B
Python
40 lines
864 B
Python
|
import os
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
current_file_path = os.path.abspath(__file__)
|
||
|
|
|
||
|
|
|
||
|
|
def run_docker_container():
|
||
|
|
user = os.getenv("USER")
|
||
|
|
container_name = f"gello_{user}"
|
||
|
|
gello_path = os.path.abspath(os.path.join(current_file_path, "../../"))
|
||
|
|
volume_mapping = f"{gello_path}:/gello"
|
||
|
|
|
||
|
|
cmd = [
|
||
|
|
"docker",
|
||
|
|
"run",
|
||
|
|
"--runtime=nvidia",
|
||
|
|
"--rm",
|
||
|
|
"--name",
|
||
|
|
container_name,
|
||
|
|
"--privileged",
|
||
|
|
"--volume",
|
||
|
|
volume_mapping,
|
||
|
|
"--volume",
|
||
|
|
"/home/gello:/homefolder",
|
||
|
|
"--net=host",
|
||
|
|
"--volume",
|
||
|
|
"/dev/serial/by-id/:/dev/serial/by-id/",
|
||
|
|
"-it",
|
||
|
|
"gello:latest",
|
||
|
|
"bash",
|
||
|
|
"-c",
|
||
|
|
"pip install -e third_party/DynamixelSDK/python && exec bash",
|
||
|
|
]
|
||
|
|
|
||
|
|
subprocess.run(cmd)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
run_docker_container()
|