nvidia-docker - Githubissues

merantix / imitation-learning

Autonomous driving: Tensorflow implementation of the paper "End-to-end Driving via Conditional Imitation Learning"

MIT License

89 stars 21 forks source link

Closed zxbnjust closed 5 years ago

zxbnjust commented 5 years ago

Is it necessary to inatall nvidia-docker?Or just need to run "docker build --build-arg base_image=tensorflow/tensorflow:1.12.0-gpu -t imit-learn . "

zxbnjust commented 5 years ago

It's necessary and we should use nvidia-docker run to train the model.