Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Dockerfile with basic requirements #10

Merged
merged 6 commits into from
Oct 8, 2019
Merged

Dockerfile with basic requirements #10

merged 6 commits into from
Oct 8, 2019

Conversation

ShreyasSkandanS
Copy link
Contributor

This is potentially useful to others trying to quickly get set up with this repository and doesn't necessarily need to be merged with master unless the authors see it fit,

I've created a quick setup Dockerfile that uses CUDA 10.0, PyTorch 1.0.0, CuDNN 7.0 and installs all the necessary dependencies to at least run the evaluation script (training untested).

The evaluation script would need to be run with Python3.5 if you use this Dockerfile.

python3.5 train.py --evaluate --snapshot checkpoints/best_cityscapes_checkpoint.pth

I've tested it on a DGX station that's using the NVIDIA 418.67 driver and it works.

Hopefully this is useful to others.

Best regards,
Shreyas

Add default epoch for evaluation script
Added some more of the requirements not mentioned in the original README.
Added more requirements, added temporary fix for evaluation script (#9)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants