saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Revert nightly docker build cuda version to 11.1.1. (#52234)

Browse source 
Summary:
CUDA 11.2 has a performance regression, so revert to CUDA 11.1.1.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/52234

Test Plan: [CI](https://github.com/pytorch/pytorch/actions?query=workflow%3A%22Build+PyTorch+nightly+Docker+image+and+push+to+GitHub+Container+Registry%22)

Reviewed By: glaringlee

Differential Revision: D26519105

Pulled By: xuzhao9

fbshipit-source-id: d1e1ecb7904c196292d83767b71000b465de73ce
This commit is contained in:
Xu Zhao 2021-02-18 12:25:01 -08:00 committed by Facebook GitHub Bot
parent 49c90648d3
commit 2d4354423e
1 changed files with 15 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
.github/scripts/build_publish_nightly_docker.sh vendored
View file

@ -3,13 +3,13 @@
set -xeuo pipefail
PYTORCH_DOCKER_TAG=$(git describe --tags --always)-devel
CUDA_VERSION=11.1.1
# Build PyTorch nightly docker
# Full name: ghcr.io/pytorch/pytorch-nightly:${PYTORCH_DOCKER_TAG}
make -f docker.Makefile \
DOCKER_REGISTRY=ghcr.io \
DOCKER_ORG=pytorch \
CUDA_VERSION=11.2.0 \
CUDA_VERSION=${CUDA_VERSION} \
CUDA_CHANNEL=conda-forge \
DOCKER_IMAGE=pytorch-nightly \
DOCKER_TAG=${PYTORCH_DOCKER_TAG} \
@ -19,8 +19,12 @@ make -f docker.Makefile \
PYTORCH_NIGHTLY_COMMIT=$(docker run \
ghcr.io/pytorch/pytorch-nightly:${PYTORCH_DOCKER_TAG} \
python -c 'import torch; print(torch.version.git_version)' | head -c 7)
docker tag ghcr.io/pytorch/pytorch-nightly:${PYTORCH_DOCKER_TAG} \
ghcr.io/pytorch/pytorch-nightly:${PYTORCH_NIGHTLY_COMMIT}
ghcr.io/pytorch/pytorch-nightly:${PYTORCH_NIGHTLY_COMMIT}-cu${CUDA_VERSION}
docker tag ghcr.io/pytorch/pytorch-nightly:${PYTORCH_NIGHTLY_COMMIT}-cu${CUDA_VERSION} \
ghcr.io/pytorch/pytorch-nightly:latest
# Push the nightly docker to GitHub Container Registry
echo $GHCR_PAT | docker login ghcr.io -u pytorch --password-stdin
@ -28,5 +32,12 @@ make -f docker.Makefile \
DOCKER_REGISTRY=ghcr.io \
DOCKER_ORG=pytorch \
DOCKER_IMAGE=pytorch-nightly \
DOCKER_TAG=${PYTORCH_NIGHTLY_COMMIT} \
DOCKER_TAG=${PYTORCH_NIGHTLY_COMMIT}-cu${CUDA_VERSION} \
devel-push
make -f docker.Makefile \
DOCKER_REGISTRY=ghcr.io \
DOCKER_ORG=pytorch \
DOCKER_IMAGE=pytorch-nightly \
DOCKER_TAG=latest \
devel-push