Skip to content
Snippets Groups Projects
Commit a65b1141 authored by Austin Anderson's avatar Austin Anderson
Browse files

Update runtime containers to Ubuntu 22.04, Python 3.11

parent 249b2413
Branches
No related tags found
No related merge requests found
# Copyright 2023 The TensorFlow Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ============================================================================
name: Continuous job tester
on:
schedule:
- cron: '*/5 * * * *'
workflow_dispatch:
jobs:
dashboards:
if: github.repository == 'angerson/build' # Don't do this yet
name: A continuous job
runs-on: ubuntu-latest
permissions:
statuses: write
strategy:
matrix:
config: [ tensorflow, jax ]
steps:
- uses: myrotvorets/set-commit-status-action@master
with:
status: ${{ job.status }}
sha: ${{ github.sha }}
......@@ -13,7 +13,7 @@
# limitations under the License.
# ============================================================================
FROM ubuntu:20.04 as base
FROM ubuntu:22.04 as base
ENV DEBIAN_FRONTEND=noninteractive
ENV LANG C.UTF-8
......@@ -24,7 +24,7 @@ RUN /setup.sources.sh
RUN /setup.packages.sh /cpu.packages.txt
ARG PYTHON_VERSION=python3.8
ARG PYTHON_VERSION=python3.11
ARG TENSORFLOW_PACKAGE=tf-nightly
COPY setup.python.sh /setup.python.sh
COPY cpu.requirements.txt /cpu.requirements.txt
......
......@@ -13,7 +13,7 @@
# limitations under the License.
# ============================================================================
FROM nvidia/cuda:11.8.0-base-ubuntu20.04 as base
FROM nvidia/cuda:11.8.0-base-ubuntu22.04 as base
ENV DEBIAN_FRONTEND=noninteractive
ENV LANG C.UTF-8
......@@ -23,7 +23,7 @@ COPY gpu.packages.txt /gpu.packages.txt
RUN /setup.sources.sh
RUN /setup.packages.sh /gpu.packages.txt
ARG PYTHON_VERSION=python3.8
ARG PYTHON_VERSION=python3.11
ARG TENSORFLOW_PACKAGE=tf-nightly
COPY setup.python.sh /setup.python.sh
COPY gpu.requirements.txt /gpu.requirements.txt
......
......@@ -13,9 +13,8 @@ libcusparse-11-8
libcublas-11-8
# CuDNN: https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#ubuntu-network-installation
libcudnn8=8.6.0.163-1+cuda11.8
# TensorRT: See https://docs.nvidia.com/deeplearning/sdk/tensorrt-install-guide/index.html#maclearn-net-repo-install-rpm
libnvinfer-plugin8=8.4.3-1+cuda11.6
libnvinfer8=8.4.3-1+cuda11.6
libnvinfer-plugin8=8.6.1.6-1+cuda11.8
libnvinfer8=8.6.1.6-1+cuda11.8
# Other packages
build-essential
......
......@@ -25,15 +25,9 @@ export DEBIAN_FRONTEND=noninteractive
# Set up shared custom sources
apt-get update
apt-get install -y gnupg ca-certificates
apt-get install -y gnupg ca-certificates wget
# Explicitly request Nvidia repo keys
# See: https://forums.developer.nvidia.com/t/invalid-public-key-for-cuda-apt-repository/212901/11
apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
# Set up custom sources
cat >/etc/apt/sources.list.d/custom.list <<SOURCES
# Nvidia CUDA packages: 18.04 has more available than 20.04, and we use those
deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /
SOURCES
# Install Nvidia repo keys
# See: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#network-repo-installation-for-ubuntu
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
dpkg -i cuda-keyring_1.1-1_all.deb
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment