Update Dockerfile.tmpl

Author: djherbis

Dockerfile.tmpl

@@ -109,7 +109,7 @@ RUN mamba install implicit && \
 
 # Install PyTorch
 {{ if eq .Accelerator "gpu" }}
-#COPY --from=torch_whl /tmp/whl/*.whl /tmp/torch/
+COPY --from=torch_whl /tmp/whl/*.whl /tmp/torch/
 RUN mamba install -c pytorch magma-cuda${CUDA_MAJOR_VERSION}${CUDA_MINOR_VERSION} && \
     pip install /tmp/torch/*.whl && \
     # b/255757999 openmp (libomp.so) is an dependency of libtorchtext and libtorchaudio but