Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[ci] Upgrade conda cudatoolkit version to 11.3 #6070

Merged
merged 2 commits into from
Sep 15, 2022
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions .github/workflows/release.yml
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ jobs:
mkdir -m777 shared
docker create --user dev --name taichi_build --gpus all -v /tmp/.X11-unix:/tmp/.X11-unix \
-e DISPLAY -e PY -e GPU_BUILD -e TAICHI_CMAKE_ARGS -e PROJECT_NAME \
registry.taichigraphics.com/taichidev-ubuntu18.04:v0.3.2 \
registry.taichigraphics.com/taichidev-ubuntu18.04:v0.3.4 \
/home/dev/taichi/.github/workflows/scripts/unix_build.sh
tar -cf - ../${{ github.event.repository.name }} --mode u=+rwx,g=+rwx,o=+rwx --owner 1000 --group 1000 | docker cp - taichi_build:/home/dev/
docker start -a taichi_build
Expand All @@ -111,7 +111,7 @@ jobs:
run: |
docker create --user dev --name taichi_test --gpus all -v /tmp/.X11-unix:/tmp/.X11-unix \
-e DISPLAY -e PY -e GPU_TEST \
registry.taichigraphics.com/taichidev-ubuntu18.04:v0.3.2 \
registry.taichigraphics.com/taichidev-ubuntu18.04:v0.3.4 \
/home/dev/unix_test.sh
docker cp .github/workflows/scripts/unix_test.sh taichi_test:/home/dev/unix_test.sh
docker cp .github/workflows/scripts/common-utils.sh taichi_test:/home/dev/common-utils.sh
Expand Down
6 changes: 3 additions & 3 deletions .github/workflows/testing.yml
Original file line number Diff line number Diff line change
Expand Up @@ -405,7 +405,7 @@ jobs:
--gpus 'all,"capabilities=graphics,utility,display,video,compute"' \
-v /tmp/.X11-unix:/tmp/.X11-unix \
-e PY -e GPU_BUILD -e PROJECT_NAME -e TAICHI_CMAKE_ARGS -e DISPLAY \
registry.taichigraphics.com/taichidev-ubuntu18.04:v0.3.3 \
registry.taichigraphics.com/taichidev-ubuntu18.04:v0.3.4 \
/home/dev/taichi/.github/workflows/scripts/unix_build.sh
# A tarball is needed because sccache needs some permissions that only the file owner has.
# 1000 is the uid and gid of user "dev" in the container.
Expand Down Expand Up @@ -444,7 +444,7 @@ jobs:
-e TI_LITE_TEST \
-e TI_TEST_OFFLINE_CACHE \
-e DISPLAY -e PY -e GPU_TEST -e TI_WANTED_ARCHS -e TI_RUN_RELEASE_TESTS \
registry.taichigraphics.com/taichidev-ubuntu18.04:v0.3.3 \
registry.taichigraphics.com/taichidev-ubuntu18.04:v0.3.4 \
/home/dev/unix_test.sh
docker cp .github/workflows/scripts/unix_test.sh taichi_test:/home/dev/unix_test.sh
docker cp .github/workflows/scripts/common-utils.sh taichi_test:/home/dev/common-utils.sh
Expand Down Expand Up @@ -671,7 +671,7 @@ jobs:
docker run --user dev --name taichi_build_host \
$DOCKER_RUN_ARGS \
-v $TAICHI_WHEEL_DIR:/home/dev/taichi/dist \
registry.taichigraphics.com/taichidev-ubuntu18.04:v0.3.3 \
registry.taichigraphics.com/taichidev-ubuntu18.04:v0.3.4 \
/home/dev/taichi/.github/workflows/scripts/unix-build-v2.sh
env:
TAICHI_CMAKE_ARGS: >-
Expand Down
18 changes: 9 additions & 9 deletions ci/Dockerfile.ubuntu.18.04
Original file line number Diff line number Diff line change
Expand Up @@ -93,17 +93,17 @@ RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh &
bash Miniconda3-latest-Linux-x86_64.sh -p /home/dev/miniconda -b
ENV PATH="/home/dev/miniconda/bin:$PATH"

# Set up multi-python environment
RUN conda init bash
RUN conda create -n py36 python=3.6 pytorch cudatoolkit=10.2 -c pytorch -y
RUN conda create -n py37 python=3.7 pytorch cudatoolkit=10.2 -c pytorch -y
RUN conda create -n py38 python=3.8 pytorch cudatoolkit=10.2 -c pytorch -y
RUN conda create -n py39 python=3.9 pytorch cudatoolkit=10.2 -c pytorch -y
# TODO add torch to 3.10 when supported
RUN conda create -n py310 python=3.10 -y

# Remove mesa EGL driver, which interferes with the propritary NVIDIA drivers
RUN rm -f /usr/lib/x86_64-linux-gnu/libEGL_mesa*

WORKDIR /home/dev
ENV LANG="C.UTF-8"

# Set up multi-python environment
RUN conda init bash
RUN conda create -n py36 python=3.6 pytorch cudatoolkit=11.3 -c pytorch -y
RUN conda create -n py37 python=3.7 pytorch cudatoolkit=11.3 -c pytorch -y
RUN conda create -n py38 python=3.8 pytorch cudatoolkit=11.3 -c pytorch -y
RUN conda create -n py39 python=3.9 pytorch cudatoolkit=11.3 -c pytorch -y
# TODO add torch to 3.10 when supported
RUN conda create -n py310 python=3.10 -y