mirror of https://github.com/hpcaitech/ColossalAI
[misc] Update PyTorch version in docs (#5711)
Co-authored-by: Edenzzzz <wtan45@wisc.edu>pull/5719/head
parent
537f6a3855
commit
785cd9a9c9
|
@ -1,7 +1,7 @@
|
|||
# Setup
|
||||
|
||||
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)
|
||||
|
|
|
@ -2,7 +2,7 @@
|
|||
|
||||
环境要求:
|
||||
|
||||
- PyTorch >= 1.11 并且 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)
|
||||
|
|
Loading…
Reference in New Issue