mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
Update PyTorch docker base image to Ubuntu-20.04 (take 2) (#101310)
Followup after https://github.com/pytorch/pytorch/pull/101301 <!-- copilot:poem --> ### <samp>🤖 Generated by Copilot at 219c58a</samp> > _`BASE_RUNTIME` changed_ > _Ubuntu twenty oh four_ > _Spring of new features_ Pull Request resolved: https://github.com/pytorch/pytorch/pull/101310 Approved by: https://github.com/atalman
This commit is contained in:
committed by
PyTorch MergeBot
parent
5fe834afc1
commit
47c99e3a1c
@ -10,8 +10,8 @@ endif
|
||||
|
||||
CUDA_VERSION = 11.7.0
|
||||
CUDNN_VERSION = 8
|
||||
BASE_RUNTIME = ubuntu:18.04
|
||||
BASE_DEVEL = nvidia/cuda:$(CUDA_VERSION)-cudnn$(CUDNN_VERSION)-devel-ubuntu18.04
|
||||
BASE_RUNTIME = ubuntu:20.04
|
||||
BASE_DEVEL = nvidia/cuda:$(CUDA_VERSION)-cudnn$(CUDNN_VERSION)-devel-ubuntu20.04
|
||||
CMAKE_VARS ?=
|
||||
|
||||
# The conda channel to use to install cudatoolkit
|
||||
|
Reference in New Issue
Block a user