Toolkit for running PyTorch training scripts on SageMaker. Dockerfiles used for building SageMaker Pytorch Containers are at https://github.com/aws/deep-learning-containers.
Apache License 2.0
197
stars
87
forks
source link
Dockerfile installation of torch and torchvision from s3, replacing original versions. #228
What did you find confusing?
In the Dockerfile.gpu, there is a point where torch and torchvision are uni-nstalled, to be replaced with the re-installed specialized version of both packages from:
Describe how documentation can be improved
Why are these replaced? During image building, this process is time-consuming. What what happen if I removed these versions and kept the original?
What did you find confusing? In the Dockerfile.gpu, there is a point where torch and torchvision are uni-nstalled, to be replaced with the re-installed specialized version of both packages from:
Around line 138, in the Dockerfile.
Describe how documentation can be improved Why are these replaced? During image building, this process is time-consuming. What what happen if I removed these versions and kept the original?