Skip to content

Commit

Permalink
[Dev] update CI config (#3377)
Browse files Browse the repository at this point in the history
  • Loading branch information
xiexinch authored Oct 12, 2023
1 parent 64172d6 commit 19b0e90
Show file tree
Hide file tree
Showing 2 changed files with 27 additions and 9 deletions.
2 changes: 2 additions & 0 deletions .circleci/docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@ ARG CUDNN="7"

FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel

ARG DEBIAN_FRONTEND=noninteractive

# To fix GPG key error when running apt-get update
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
Expand Down
34 changes: 25 additions & 9 deletions .circleci/test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -82,16 +82,22 @@ jobs:
type: string
cuda:
type: enum
enum: ["10.1", "10.2", "11.1"]
enum: ["10.1", "10.2", "11.1", "11.7", "11.8"]
cudnn:
type: integer
default: 7
machine:
image: ubuntu-2004-cuda-11.4:202110-01
# docker_layer_caching: true
resource_class: gpu.nvidia.small
image: linux-cuda-11:default
docker_layer_caching: true
resource_class: gpu.nvidia.small.multi
steps:
- 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:
# Cloning repos in VM since Docker doesn't have access to the private key
name: Clone Repos
Expand Down Expand Up @@ -154,9 +160,8 @@ workflows:
- lint
- build_cpu:
name: maximum_version_cpu
# TODO: Fix torch 1.13 forward crush
torch: 1.12.0
torchvision: 0.13.0
torch: 2.0.0
torchvision: 0.15.0
python: 3.9.0
requires:
- minimum_version_cpu
Expand All @@ -166,10 +171,10 @@ workflows:
- maximum_version_cpu
- build_cuda:
name: mainstream_version_gpu
torch: 1.8.1
torch: 2.0.0
# Use double quotation mark to explicitly specify its type
# as string instead of number
cuda: "10.2"
cuda: "11.8"
requires:
- hold
merge_stage_test:
Expand All @@ -188,3 +193,14 @@ workflows:
only:
- dev-1.x
- main
- build_cuda:
name: maximum_version_gpu
torch: 2.0.0
# Use double quotation mark to explicitly specify its type
# as string instead of number
cuda: "11.8"
cudnn: 8
filters:
branches:
only:
- main

0 comments on commit 19b0e90

Please sign in to comment.