File tree 3 files changed +11
-3
lines changed
3 files changed +11
-3
lines changed Original file line number Diff line number Diff line change @@ -2,6 +2,8 @@ ARG PYTORCH="1.8.1"
2
2
ARG CUDA="10.2"
3
3
ARG CUDNN="7"
4
4
5
+ ARG DEBIAN_FRONTEND=noninteractive
6
+
5
7
FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
6
8
7
9
# 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:
90
90
type : integer
91
91
default : 8
92
92
machine :
93
- image : ubuntu-2004- cuda-11.4:202110-01
93
+ image : linux- cuda-11:default
94
94
# docker_layer_caching: true
95
- resource_class : gpu.nvidia.small
95
+ resource_class : gpu.nvidia.small.multi
96
96
steps :
97
97
- 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
98
104
- run :
99
105
# Cloning repos in VM since Docker doesn't have access to the private key
100
106
name : Clone Repos
Original file line number Diff line number Diff line change 1
1
# Copyright (c) Open-MMLab. All rights reserved.
2
2
3
- __version__ = '1.2 .0'
3
+ __version__ = '1.3 .0'
4
4
short_version = __version__
5
5
6
6
You can’t perform that action at this time.
0 commit comments