【trition-server】运行一个pytorch的ngc镜像

时间:2024-07-05 21:24:23

  • ngc 提供了pytorch容器

  • 号称是做了gpu加速的

  • 我装的系统版本是3.8的python,但是pytorch似乎是用conda安装的3.5的:

torch的python库是ls支持gpu加速是真的

在这里插入图片描述

英伟达的pytorch的说明书

root@a79bc3874b9d:/opt/pytorch# cat NVREADME.md

PyTorch

=======

  • PyTorch is a python package that provides two high-level features:

*** Tensor computation (like numpy) with strong GPU acceleration

  • Deep Neural Networks built on a tape-based autograd system**

You can reuse your favorite python packages such as numpy, scipy and Cython to
extend PyTorch when needed.

Contents of the PyTorc