sagemaker-pytorch-training-toolkit icon indicating copy to clipboard operation
sagemaker-pytorch-training-toolkit copied to clipboard

Dockerfile installation of torch and torchvision from s3, replacing original versions.

Open akinolawilson opened this issue 4 years ago • 0 comments

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?

akinolawilson avatar Mar 07 '21 15:03 akinolawilson