深度å¦ä¹ æ¡æ¶
深度å¦ä¹ (DL) æ¡æ¶æä¾äºéè¿é«çº§ç¼ç¨æ¥å£è®¾è®¡ãè®ç»åéªè¯æ·±åº¦ç¥ç»ç½ç»çæå»ºåã广æ³ä½¿ç¨ç深度å¦ä¹ æ¡æ¶ï¼ä¾å¦ PyTorchãTensorFlowã PyTorch GeometricãDGLçï¼ä¾èµ GPU å éåºï¼ä¾å¦ cuDNNãNCCL å DALIï¼æ¥æä¾é«æ§è½ãå¤ GPU å éçè®ç»ã
æ¥çæææ¡æ¶
NVIDIA ä¼åçæ·±åº¦å¦ä¹ æ¡æ¶
å¼å人åãç 究人ååæ°æ®ç§å¦å®¶å¯ä»¥è½»æ¾è®¿é® NVIDIA AI ä¼åçæ·±åº¦å¦ä¹ æ¡æ¶å®¹å¨ï¼å ¶ä¸å å«é对 NVIDIA GPU è¿è¡æ§è½è°æ´åæµè¯ç深度å¦ä¹ 示ä¾ãè¿æ¶é¤äºç®¡çå åä¾èµé¡¹æä»æºä»£ç æå»º DL æ¡æ¶çéè¦ãå 嫿æä¾èµé¡¹ç容å¨å深度å¦ä¹ æ¡æ¶æä¾äºä¸ä¸ªè½»æ¾å¼å§å¼å常è§åºç¨ç¨åºçå°æ¹ï¼ä¾å¦å¯¹è¯å¼ AI ãèªç¶è¯è¨çè§£ (NLU)ãæ¨èç³»ç»åè®¡ç®æºè§è§ã请访é®NVIDIA NGC⢠ç®å½ä»¥äºè§£æ´å¤ä¿¡æ¯ã
PyTorch
PyTorch æ¯ä¸ä¸ª Python å ï¼æä¾ä¸¤ä¸ªé«çº§åè½ï¼
- å ·æå¼ºå¤§ GPU å éåè½çå¼ é计ç®ï¼å¦ numpyï¼ã
- 深度ç¥ç»ç½ç» (DNN) æå»ºå¨åºäºç£å¸¦çèªå¨å级系ç»ä¸ã
å¨éè¦æ¶éç¨æ¨æå欢ç Python å ï¼ä¾å¦ numpyãscipy å Cythonï¼æ¥æ©å± PyTorchã
NGC ä¸ç PyTorchç¤ºä¾æ¨¡åèªå¨æ··å精度
模åé¨ç½²
å¯¹äº PyTorch è®ç»æ¨¡åç髿§è½æ¨çé¨ç½²ï¼
- 1. ä½¿ç¨ Torch-TensorRT éæå¨ PyTorch ä¸ä¼ååé¨ç½²æ¨¡åã
- 2. å° PyTorch 模å导åºä¸ºONNXæ ¼å¼ï¼å¹¶ä½¿ç¨NVIDIA TensorRTâ¢ï¼ç¨äºé«æ§è½æ·±åº¦å¦ä¹ æ¨çç SDKï¼å¯¼å ¥ãä¼ååé¨ç½²ã
å¦ä¹ èµæº
- 深度å¦ä¹ ææ¡£ï¼ ä» NGC ç®å½è¿è¡ PyTorch 容å¨
- PyTorch æç¨
- Torch-TensorRT ææ¡£
TensorFlow
TensorFlow æ¯ä¸ä¸ªä½¿ç¨æ°æ®æµå¾è¿è¡æ°å¼è®¡ç®ç弿ºè½¯ä»¶åºãå¾ä¸çèç¹è¡¨ç¤ºæ°å¦è¿ç®ï¼èå¾ç边表示å¨å®ä»¬ä¹é´æµå¨çå¤ç»´æ°æ®æ°ç»ï¼å¼ éï¼ãè¿ç§çµæ´»çæ¶æå 许æ¨å°è®¡ç®é¨ç½²å°æ¡é¢ãæå¡å¨æç§»å¨è®¾å¤ä¸çä¸ä¸ªæå¤ä¸ª CPU æ GPUï¼èæ ééå代ç ã为äºå¯è§å TensorFlow ç»æï¼TensorFlow æä¾äº TensorBoardï¼è¿æ¯ä¸å¥å¯è§åå·¥å ·ã
NGC ä¸ç TensorFlow GitHub ä¸ç TensorFlowç¤ºä¾æ¨¡åéç¨äº JetPack çèªå¨æ··å精度TensorFlow
模åé¨ç½²
å¯¹äº TensorFlow è®ç»æ¨¡åç髿§è½æ¨çé¨ç½²ï¼
- 1. ä½¿ç¨ TensorFlow-TensorRT éæå¨ TensorFlow ä¸ä¼ååé¨ç½²æ¨¡åã
- 2. å° TensorFlow 模å导åºå°ONNX ï¼å¹¶ä½¿ç¨NVIDIA TensorRTï¼ç¨äºé«æ§è½æ·±åº¦å¦ä¹ æ¨çç SDKï¼å¯¼å ¥ãä¼ååé¨ç½²ã
å¦ä¹ èµæº
- 深度å¦ä¹ ææ¡£ï¼TensorFlow ç¨æ·æå
- 深度å¦ä¹ ææ¡£ï¼ TensorFlow æä½³å®è·µ
- TensorFlow å ¥é¨æå
JAX
JAX æ¯ä¸ä¸ªä¸ä¸ºé«æ§è½æ°å¼è®¡ç®åæºå¨å¦ä¹ ç ç©¶è设计ç Python åºã JAXå¯ä»¥èªå¨åºååçPythonå¹¶å®ç°NumPy APIãåªéæ´æ¹å è¡ä»£ç ï¼JAX å³å¯å®ç°è·¨å¤èç¹ãå¤ GPU ç³»ç»çåå¸å¼è®ç»ï¼å¹¶éè¿ NVIDIA GPU ä¸ç XLA ä¼åå æ ¸å éæ§è½ã Python å NumPy é½è¢«å¹¿æ³ä½¿ç¨åçæï¼ä½¿å¾ JAX ç®åãçµæ´»ä¸æäºéç¨ã
ç³è¯·æ©æè®¿é®å¼åè 论åç½ç»ç 讨ä¼
PaddlePaddle
PaddlePaddle æä¾äºä¸ä¸ªç´è§ä¸çµæ´»çç颿¥å è½½æ°æ®åæå®æ¨¡åç»æãæ¯æCNNãRNNçå¤ç§åä½ï¼è½»æ¾é ç½®å¤æçæ·±åº¦æ¨¡åã
PaddlePaddleè¿æä¾æå ¶ä¼åçæä½ãå ååæ¶åç½ç»éä¿¡ï¼å¹¶å¯ä»¥è½»æ¾æ©å±å¼æè®¡ç®èµæºååå¨ä»¥å éè®ç»è¿ç¨ã
NGC PaddlePaddle å®è£ 页é¢ä¸ç PaddlePaddle PADDLEPADDLE æº
模åé¨ç½²
å¯¹äº PaddlePaddle è®ç»æ¨¡åç髿§è½æ¨çé¨ç½²ï¼
- ä½¿ç¨ Paddle-TensorRT éæå¨ PaddlePaddle ä¸ä¼ååé¨ç½²æ¨¡åã
- å° PaddlePaddle 模å导åºå° ONNXï¼å¹¶ä½¿ç¨ NVIDIA TensorRTï¼ä¸ç§ç¨äºé«æ§è½æ·±åº¦å¦ä¹ æ¨çç SDKï¼å¯¼å ¥ãä¼ååé¨ç½²ã
å¦ä¹ èµæº
MXNet
MXNet æ¯ä¸ä¸ªä¸ä¸ºæé«æçåçµæ´»æ§èè®¾è®¡çæ·±åº¦å¦ä¹ æ¡æ¶ãå®å è®¸æ¨æ··å符å·ç¼ç¨åå½ä»¤å¼ç¼ç¨ç飿 ¼ï¼ä»¥æå¤§éåº¦å°æé«æçåç产åã
å ¶æ ¸å¿æ¯ä¸ä¸ªå¨æä¾èµè°åº¦ç¨åºï¼å¯ä»¥å¨æèªå¨å¹¶è¡åç¬¦å·æä½åå½ä»¤æä½ãå ¶ä¹ä¸çå¾å½¢ä¼åå±ä½¿ç¬¦å·æ§è¡å¿«éä¸å å髿ãè¯¥åºæ¯å¯ç§»æ¤çãè½»é级çï¼å¹¶ä¸å¯ä»¥æ©å±å°å¤ä¸ª GPU åæºå¨ã
NGC ä¸ç MXNetæ ·æ¬æ¨¡åèªå¨æ··å精度
模åé¨ç½²
对äºç»è¿è®ç»ç模åç髿§è½æ¨çé¨ç½²ï¼å¯¼åºä¸ºONNXæ ¼å¼ï¼å¹¶ä½¿ç¨NVIDIA TensorRT ï¼ä¸ç§ç¨äºé«æ§è½æ·±åº¦å¦ä¹ æ¨çç SDKï¼è¿è¡ä¼ååé¨ç½²ã
å¦ä¹ èµæº
- 深度å¦ä¹ ææ¡£ï¼ è¿è¡ MXNet 容å¨
MATLAB
MATLAB 让工ç¨å¸ãç§å¦å®¶åé¢åä¸å®¶è½å¤è½»æ¾è¿è¡æ·±åº¦å¦ä¹ ã MATLAB ä¸ä» å ·æç¨äºç®¡çåæ è®°å¤§åæ°æ®éçå·¥å ·ååè½ï¼è¿æä¾ç¨äºå¤çæºå¨å¦ä¹ ãç¥ç»ç½ç»ãè®¡ç®æºè§è§åèªå¨é©¾é©¶çä¸ç¨å·¥å ·ç®±ãåªéå è¡ä»£ç ï¼MATLAB å°±å¯ä»¥è®©æ¨å建åå¯è§å模åï¼å¹¶å°æ¨¡åé¨ç½²å°æå¡å¨ååµå ¥å¼è®¾å¤ï¼èæ éæä¸ºä¸å®¶ã MATLAB è¿ä½¿ç¨æ·è½å¤ä» MATLAB 代ç èªå¨çæç¨äº DL åè§è§åºç¨ç¨åºç髿§è½ CUDA 代ç ã
NGC ä¸ç MATLABç¨äºæ·±åº¦å¦ä¹ ç MATLAB
模åé¨ç½²
为äºå¯¹ MATLAB è®ç»æ¨¡åè¿è¡é«æ§è½æ¨çé¨ç½²ï¼è¯·ä½¿ç¨MATLAB GPU Coderèªå¨çæä»äºå°åµå ¥å¼é¨ç½²ç¯å¢ç TensorRT ä¼åæ¨ç弿ã
å¦ä¹ èµæº




