"...text-generation-inference.git" did not exist on "0204946d26d4790c230c4aa4b35e261fd598fd5f"
Commit be3a6c7c authored by sunjiahao1999's avatar sunjiahao1999
Browse files

fix version and resource_class & image name

parent 12b595ca
...@@ -2,6 +2,8 @@ ARG PYTORCH="1.8.1" ...@@ -2,6 +2,8 @@ ARG PYTORCH="1.8.1"
ARG CUDA="10.2" ARG CUDA="10.2"
ARG CUDNN="7" ARG CUDNN="7"
ARG DEBIAN_FRONTEND=noninteractive
FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
# To fix GPG key error when running apt-get update # To fix GPG key error when running apt-get update
......
...@@ -90,11 +90,17 @@ jobs: ...@@ -90,11 +90,17 @@ jobs:
type: integer type: integer
default: 8 default: 8
machine: machine:
image: ubuntu-2004-cuda-11.4:202110-01 image: linux-cuda-11:default
# docker_layer_caching: true # docker_layer_caching: true
resource_class: gpu.nvidia.small resource_class: gpu.nvidia.small
steps: steps:
- checkout - checkout
- run:
name: Install nvidia-container-toolkit and Restart Docker
command: |
sudo apt-get update
sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker
- run: - run:
# Cloning repos in VM since Docker doesn't have access to the private key # Cloning repos in VM since Docker doesn't have access to the private key
name: Clone Repos name: Clone Repos
......
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) Open-MMLab. All rights reserved.
__version__ = '1.2.0' __version__ = '1.3.0'
short_version = __version__ short_version = __version__
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment