PyTorch 1.11CUDA 11.3(在 x86_64 上,默认设置下,兼容的 PyTorch Build 支持较新版本的 CUDA 11)cuDNN 8.2.4.15TensorRT 8.2.4.2由于最近 JetPack 的升级和其发布流程的变化,Torch-TRTv1.1.0 版本不支持Jetson(Jetpack 5.0DP 或其他)。将在中期版本(Torch-TensorRT1.1.x)同时支持 Jetpack 5.0DP 和 TensorRT 8....
1. 参与深度学习框架OneFlow开发与性能优化; 2. 参与OneFlow CPU/GPU高性能算子库研发 ; 3. 负责深度学习框架OneFlow接口设计。 职位要求: 1. 计算机或电子通信相关专业本科以上; 2. 熟练掌握C++、Python; 3. 熟悉GPU(CUDA、cuDNN、NCCL)编程; 4 良好的软件开发素养,包括TDD、CI/CD、敏捷开发流程等; ...