OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 00:20:18 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Use the conda channel defined in docker.Makefile to install cudatoolkit (#53316)

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/53316

Test Plan:
Nightly Docker build CI

This is a follow-up PR after docker moved default CUDA => 11.1. Only merge this after https://github.com/pytorch/pytorch/issues/53299 is committed.

Reviewed By: albanD

Differential Revision: D26996287

Pulled By: xuzhao9

fbshipit-source-id: 0c2e03da41d036d7aada3e07d479a3dede219f58
This commit is contained in:
Xu Zhao 2021-03-12 11:13:49 -08:00 committed by Facebook GitHub Bot
parent 90dfdef226
commit 7763bb6cb3
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.github/scripts/build_publish_nightly_docker.sh vendored
View File

@ -3,14 +3,13 @@
set -xeuo pipefail
PYTORCH_DOCKER_TAG=$(git describe --tags --always)-devel
CUDA_VERSION=11.1.1
CUDA_VERSION=11.1
# Build PyTorch nightly docker
make -f docker.Makefile \
DOCKER_REGISTRY=ghcr.io \
DOCKER_ORG=pytorch \
CUDA_VERSION=${CUDA_VERSION} \
CUDA_CHANNEL=conda-forge \
DOCKER_IMAGE=pytorch-nightly \
DOCKER_TAG=${PYTORCH_DOCKER_TAG} \
INSTALL_CHANNEL=pytorch-nightly BUILD_TYPE=official devel-image

2
docker.Makefile
View File

@ -14,7 +14,7 @@ BASE_RUNTIME = ubuntu:18.04
BASE_DEVEL = nvidia/cuda:$(CUDA_VERSION)-cudnn$(CUDNN_VERSION)-devel-ubuntu18.04
# The conda channel to use to install cudatoolkit
CUDA_CHANNEL = defaults
CUDA_CHANNEL = nvidia
# The conda channel to use to install pytorch / torchvision
INSTALL_CHANNEL = pytorch