-
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.