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
PaddlePaddle · May 21, 2021 · 44668a7 · 44668a7
1 parent 70dc5f4
commit 44668a7
Showing 1 changed file with 27 additions and 57 deletions.
diff --git a/paddle/scripts/conda_build.py b/paddle/scripts/conda_build.py
@@ -44,42 +44,33 @@ def __init__(self):
         self.requirement_build = r"""
 requirements:
   build:
-    - numpy>=1.12
+    - numpy>=1.13
     - cython
     - setuptools
 """
 
         self.requirement_run = r"""
   run:
-    - numpy>1.12
+    - requests>=2.20.0
+    - numpy>=1.13
+    - protobuf>=3.1.0
+    - gast==0.3.3
+    - Pillow
     - six
     - decorator
-    - nltk
-    - scipy
-    - requests
-    - pillow
-    - graphviz
-    - protobuf
-    - py-cpuinfo==5.0.0
     - astor
-    - gast>=0.3.3
-    - matplotlib
 """
 
         self.requirement_run_windows = r"""
   run:
-    - numpy>=1.12
+    - requests>=2.20.0
+    - numpy>=1.13
+    - protobuf>=3.1.0
+    - gast==0.3.3
+    - Pillow
     - six
     - decorator
-    - nltk
-    - scipy
-    - requests
-    - pillow
-    - graphviz
-    - protobuf
     - astor
-    - gast>=0.3.3
-    - py-cpuinfo==5.0.0
 """
         self.test = r"""
 test:
@@ -96,37 +87,20 @@ def __init__(self):
 """
 
         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""" 
-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.python37 = r"    - python>=3.7, <3.8"
         self.python38 = r"    - python>=3.8, <3.9"
+        self.python39 = r"    - python>=3.9, <3.10"
 
         self.python_version = [
-            self.python27, self.python35, self.python36, self.python37,
-            self.python38
+            self.python36, self.python37, self.python38, self.python39
         ]
 
-        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"""
     - cudatoolkit>=10.1, <10.2
     - cudnn>=7.6, <7.7
@@ -135,30 +109,31 @@ def __init__(self):
     - cudatoolkit>=10.2, <10.3
     - cudnn>=7.6, <7.7
     """
-        self.cuda_info = [(self.cuda90, "cuda9.0", ".post90"),
-                          (self.cuda100, "cuda10.0", ".post100"),
-                          (self.cuda101, "cuda10.1", ".post101"),
-                          (self.cuda102, "cuda10.2", "")]
-        self.py_str = ["py27", "py35", "py36", "py37", "py38"]
+        self.cuda112 = r"""
+    - cudatoolkit>=11.2, <11.3
+    - cudnn>=8.1, <8.2
+    """
+
+        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_prefix_linux = "pip install /package/paddlepaddle"
         self.pip_prefix_windows = r"pip install C:\package\paddlepaddle"
         self.pip_gpu = "_gpu-"
         self.pip_cpu = "-"
         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",
-            "-cp38-cp38-macosx_10_14_x86_64"
+            "-cp38-cp38-macosx_10_14_x86_64", "-cp39-cp39-macosx_10_14_x86_64"
         ]
         self.linux_pip = [
-            "-cp27-cp27mu-manylinux1_x86_64", "-cp35-cp35m-manylinux1_x86_64",
-            "-cp36-cp36m-manylinux1_x86_64", "-cp37-cp37m-manylinux1_x86_64",
-            "-cp38-cp38-manylinux1_x86_64"
+            "-cp36-cp36m-linux_x86_64", "-cp37-cp37m-linux_x86_64",
+            "-cp38-cp38-linux_x86_64", "-cp39-cp39-linux_x86_64"
         ]
         self.windows_pip = [
-            "-cp27-cp27m-win_amd64", "-cp35-cp35m-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 @@ def meta_build_windows(var,
     requirement = var.requirement_build + python_str + var.requirement_run_windows + python_str
     meta_build = var.build + build_name_str
     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):
         meta_str = meta_str + cuda_str