PYTORCH 依赖 cuda 11.8 , cuda 12.1 cpu 里 torch torchvision torchaudio对应关系_stable diffusion cuda 12.1和cuda 11.8

作者：盐析白兔 | 2024-03-25 08:46:31

踩

stable diffusion cuda 12.1和cuda 11.8

Pytorch是Facebook人工智能研究院基于Torch开发的一个开源的Python机器学习库，它提供了两项强大的功能：

1、强大的GPU加速的张量计算（NumPy)。

2、自动求导系统的深度神经网络。

cuda 11.8

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118

cuda 12.1

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu121

cpu

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cpu

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/308377