github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-12 20:54:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

[misc] Update PyTorch version in docs (#5724)

Browse Source 
* [misc] Update PyTorch version in docs

* [misc] Update PyTorch version in docs
This commit is contained in:
binmakeswell
2024-05-16 13:54:32 +08:00
committed by GitHub
parent 913c920ecc
commit 2011b1356a
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -418,7 +418,7 @@ Please visit our [documentation](https://www.colossalai.org/) and [examples](htt
## Installation
Requirements:
- PyTorch >= 1.11 and PyTorch <= 2.1
- PyTorch >= 2.1
- Python >= 3.7
- CUDA >= 11.0
- [NVIDIA GPU Compute Capability](https://developer.nvidia.com/cuda-gpus) >= 7.0 (V100/RTX20 and higher)