未验证 提交 44668a7a 编写于 作者: Y YUNSHEN XIE 提交者: GitHub

update conda build script for cuda11 (#29594)

* update conda build script for cuda11

* update conda build script

* modified wheel name

* update conda_build

* fix error

* add cudnn8.1 for cuda11.2

* fix format error
上级 70dc5f49
...@@ -44,42 +44,33 @@ build: ...@@ -44,42 +44,33 @@ build:
self.requirement_build = r""" self.requirement_build = r"""
requirements: requirements:
build: build:
- numpy>=1.12 - numpy>=1.13
- cython - cython
- setuptools - setuptools
""" """
self.requirement_run = r""" self.requirement_run = r"""
run: run:
- numpy>1.12 - requests>=2.20.0
- numpy>=1.13
- protobuf>=3.1.0
- gast==0.3.3
- Pillow
- six - six
- decorator - decorator
- nltk
- scipy
- requests
- pillow
- graphviz
- protobuf
- py-cpuinfo==5.0.0
- astor - astor
- gast>=0.3.3
- matplotlib
""" """
self.requirement_run_windows = r""" self.requirement_run_windows = r"""
run: run:
- numpy>=1.12 - requests>=2.20.0
- numpy>=1.13
- protobuf>=3.1.0
- gast==0.3.3
- Pillow
- six - six
- decorator - decorator
- nltk
- scipy
- requests
- pillow
- graphviz
- protobuf
- astor - astor
- gast>=0.3.3
- py-cpuinfo==5.0.0
""" """
self.test = r""" self.test = r"""
test: test:
...@@ -96,37 +87,20 @@ about: ...@@ -96,37 +87,20 @@ about:
""" """
self.build_const = r""" self.build_const = r"""
pip install /package/objgraph-3.4.1.tar.gz
pip install /package/rarfile-3.0.tar.gz --no-deps
""" """
self.blt_const = r""" self.blt_const = r"""
pip install C:\package\objgraph-3.4.1.tar.gz
pip install C:\package\rarfile-3.0.tar.gz --no-deps
git clone https://github.com/PaddlePaddle/recordio.git
cd recordio\python
python setup.py install
""" """
self.python27 = r" - python>=2.7, <3.0"
self.python35 = r" - python>=3.5, <3.6"
self.python36 = r" - python>=3.6, <3.7" self.python36 = r" - python>=3.6, <3.7"
self.python37 = r" - python>=3.7, <3.8" self.python37 = r" - python>=3.7, <3.8"
self.python38 = r" - python>=3.8, <3.9" self.python38 = r" - python>=3.8, <3.9"
self.python39 = r" - python>=3.9, <3.10"
self.python_version = [ self.python_version = [
self.python27, self.python35, self.python36, self.python37, self.python36, self.python37, self.python38, self.python39
self.python38
] ]
self.cuda90 = r"""
- cudatoolkit>=9.0, <9.1
- cudnn>=7.6, <7.7
"""
self.cuda100 = r"""
- cudatoolkit>=10.0, <10.1
- cudnn>=7.6, <7.7
"""
self.cuda101 = r""" self.cuda101 = r"""
- cudatoolkit>=10.1, <10.2 - cudatoolkit>=10.1, <10.2
- cudnn>=7.6, <7.7 - cudnn>=7.6, <7.7
...@@ -135,30 +109,31 @@ python setup.py install ...@@ -135,30 +109,31 @@ python setup.py install
- cudatoolkit>=10.2, <10.3 - cudatoolkit>=10.2, <10.3
- cudnn>=7.6, <7.7 - cudnn>=7.6, <7.7
""" """
self.cuda_info = [(self.cuda90, "cuda9.0", ".post90"), self.cuda112 = r"""
(self.cuda100, "cuda10.0", ".post100"), - cudatoolkit>=11.2, <11.3
(self.cuda101, "cuda10.1", ".post101"), - cudnn>=8.1, <8.2
(self.cuda102, "cuda10.2", "")] """
self.py_str = ["py27", "py35", "py36", "py37", "py38"]
self.cuda_info = [(self.cuda101, "cuda10.1", ".post101"),
(self.cuda102, "cuda10.2", ""),
(self.cuda112, "cuda11.2", ".post112")]
self.py_str = ["py36", "py37", "py38", "py39"]
self.pip_end = ".whl --no-deps" self.pip_end = ".whl --no-deps"
self.pip_prefix_linux = "pip install /package/paddlepaddle" self.pip_prefix_linux = "pip install /package/paddlepaddle"
self.pip_prefix_windows = r"pip install C:\package\paddlepaddle" self.pip_prefix_windows = r"pip install C:\package\paddlepaddle"
self.pip_gpu = "_gpu-" self.pip_gpu = "_gpu-"
self.pip_cpu = "-" self.pip_cpu = "-"
self.mac_pip = [ self.mac_pip = [
"-cp27-cp27m-macosx_10_6_intel", "-cp35-cp35m-macosx_10_6_intel",
"-cp36-cp36m-macosx_10_6_intel", "-cp37-cp37m-macosx_10_6_intel", "-cp36-cp36m-macosx_10_6_intel", "-cp37-cp37m-macosx_10_6_intel",
"-cp38-cp38-macosx_10_14_x86_64" "-cp38-cp38-macosx_10_14_x86_64", "-cp39-cp39-macosx_10_14_x86_64"
] ]
self.linux_pip = [ self.linux_pip = [
"-cp27-cp27mu-manylinux1_x86_64", "-cp35-cp35m-manylinux1_x86_64", "-cp36-cp36m-linux_x86_64", "-cp37-cp37m-linux_x86_64",
"-cp36-cp36m-manylinux1_x86_64", "-cp37-cp37m-manylinux1_x86_64", "-cp38-cp38-linux_x86_64", "-cp39-cp39-linux_x86_64"
"-cp38-cp38-manylinux1_x86_64"
] ]
self.windows_pip = [ self.windows_pip = [
"-cp27-cp27m-win_amd64", "-cp35-cp35m-win_amd64",
"-cp36-cp36m-win_amd64", "-cp37-cp37m-win_amd64", "-cp36-cp36m-win_amd64", "-cp37-cp37m-win_amd64",
"-cp38-cp38-win_amd64" "-cp38-cp38-win_amd64", "-cp39-cp39-win_amd64"
] ]
...@@ -233,12 +208,7 @@ package: ...@@ -233,12 +208,7 @@ package:
requirement = var.requirement_build + python_str + var.requirement_run_windows + python_str requirement = var.requirement_build + python_str + var.requirement_run_windows + python_str
meta_build = var.build + build_name_str meta_build = var.build + build_name_str
meta_str = package_str + meta_build + requirement meta_str = package_str + meta_build + requirement
if (python_str == var.python27 or python_str == var.python35):
meta_str = meta_str + """
- matplotlib<=2.2.4"""
else:
meta_str = meta_str + """
- matplotlib"""
if not (cuda_str == None): if not (cuda_str == None):
meta_str = meta_str + cuda_str meta_str = meta_str + cuda_str
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册