File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -2,6 +2,8 @@ ARG PYTORCH="1.8.1"
22ARG CUDA="10.2"
33ARG CUDNN="7"
44
5+ ARG DEBIAN_FRONTEND=noninteractive
6+
57FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
68
79# To fix GPG key error when running apt-get update
Original file line number Diff line number Diff line change @@ -90,11 +90,17 @@ jobs:
9090 type : integer
9191 default : 8
9292 machine :
93- image : ubuntu-2004- cuda-11.4:202110-01
93+ image : linux- cuda-11:default
9494 # docker_layer_caching: true
95- resource_class : gpu.nvidia.small
95+ resource_class : gpu.nvidia.small.multi
9696 steps :
9797 - checkout
98+ - run :
99+ name : Install nvidia-container-toolkit and Restart Docker
100+ command : |
101+ sudo apt-get update
102+ sudo apt-get install -y nvidia-container-toolkit
103+ sudo systemctl restart docker
98104 - run :
99105 # Cloning repos in VM since Docker doesn't have access to the private key
100106 name : Clone Repos
Original file line number Diff line number Diff line change 11# Copyright (c) Open-MMLab. All rights reserved.
22
3- __version__ = '1.2 .0'
3+ __version__ = '1.3 .0'
44short_version = __version__
55
66
You can’t perform that action at this time.
0 commit comments