sagemaker-pytorch-training-toolkit
sagemaker-pytorch-training-toolkit copied to clipboard
Dockerfile installation of torch and torchvision from s3, replacing original versions.
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?