62 строки
1.9 KiB
Python
62 строки
1.9 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you 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.
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
|
|
from jinja2 import Template
|
|
|
|
CUDA_VERSIONS = ['10.0', '9.0']
|
|
|
|
|
|
# Make sure that the cudnn version you set here is available
|
|
# for all the cuda versions that you want both from nvidia
|
|
# and from conda.
|
|
|
|
# These two must be in sync
|
|
CUDNN_FULL_VERSION = '7.6.0.64'
|
|
CUDNN_VERSION = '7.6.0'
|
|
|
|
|
|
condadir = os.path.dirname(sys.argv[0])
|
|
condadir = os.path.abspath(condadir)
|
|
srcdir = os.path.dirname(condadir)
|
|
|
|
|
|
with open(os.path.join(condadir, 'Dockerfile.template')) as f:
|
|
docker_template = Template(f.read())
|
|
|
|
|
|
def render_dockerfile(version):
|
|
txt = docker_template.render(cuda_version=version,
|
|
cudnn_short_version=CUDNN_VERSION,
|
|
cudnn_version=CUDNN_FULL_VERSION)
|
|
fname = os.path.join(condadir,
|
|
'../docker/Dockerfile.conda_cuda' + version.replace('.', ''))
|
|
with open(fname, 'w') as f:
|
|
f.write(txt)
|
|
return fname
|
|
|
|
|
|
if __name__ == '__main__':
|
|
build_versions = CUDA_VERSIONS
|
|
if len(sys.argv) > 1:
|
|
build_versions = sys.argv[1:]
|
|
for version in build_versions:
|
|
render_dockerfile(version)
|