deepspeed 适配的 torch版本
时间: 2023-09-18 20:09:57 浏览: 1709
torch-1.14.0+cuda11.8+sm-90+deepspeed
5星 · 资源好评率100%
`deepspeed` 目前支持的 `torch` 版本包括:
- PyTorch 1.7
- PyTorch 1.6
- PyTorch 1.5
- PyTorch 1.4
- PyTorch 1.3
其中,PyTorch 1.7 和 1.6 版本支持 CUDA 11 和 CUDA 10.2,其他版本只支持 CUDA 10.1。
如果您想要使用 `deepspeed`,需要先安装相应版本的 `torch` 和 `torchvision` 库。您可以前往 `torch` 的官方网站(https://pytorch.org/)下载和安装最新版本的 `torch` 和 `torchvision`,或者使用 `pip` 命令进行安装,例如:
```
pip install torch torchvision
```
请注意,`deepspeed` 和 `torch` 的版本要保持一致,否则可能会出现不兼容的情况。在安装 `deepspeed` 时,建议同时安装相应版本的 `torch` 和 `torchvision` 库。
阅读全文