mirror of https://github.com/hpcaitech/ColossalAI
Merge branch 'hpcaitech:feature/prefetch' into feature/prefetch
commit
c5ddf17c76
|
@ -418,7 +418,7 @@ Please visit our [documentation](https://www.colossalai.org/) and [examples](htt
|
||||||
## Installation
|
## Installation
|
||||||
|
|
||||||
Requirements:
|
Requirements:
|
||||||
- PyTorch >= 1.11 and PyTorch <= 2.1
|
- PyTorch >= 2.1
|
||||||
- Python >= 3.7
|
- Python >= 3.7
|
||||||
- CUDA >= 11.0
|
- CUDA >= 11.0
|
||||||
- [NVIDIA GPU Compute Capability](https://developer.nvidia.com/cuda-gpus) >= 7.0 (V100/RTX20 and higher)
|
- [NVIDIA GPU Compute Capability](https://developer.nvidia.com/cuda-gpus) >= 7.0 (V100/RTX20 and higher)
|
||||||
|
|
|
@ -413,7 +413,7 @@ Colossal-AI 为您提供了一系列并行组件。我们的目标是让您的
|
||||||
|
|
||||||
环境要求:
|
环境要求:
|
||||||
|
|
||||||
- PyTorch >= 1.11 并且 PyTorch <= 2.1
|
- PyTorch >= 2.1
|
||||||
- Python >= 3.7
|
- Python >= 3.7
|
||||||
- CUDA >= 11.0
|
- CUDA >= 11.0
|
||||||
- [NVIDIA GPU Compute Capability](https://developer.nvidia.com/cuda-gpus) >= 7.0 (V100/RTX20 and higher)
|
- [NVIDIA GPU Compute Capability](https://developer.nvidia.com/cuda-gpus) >= 7.0 (V100/RTX20 and higher)
|
||||||
|
|
Loading…
Reference in New Issue