File tree Expand file tree Collapse file tree 2 files changed +5
-6
lines changed
Expand file tree Collapse file tree 2 files changed +5
-6
lines changed Original file line number Diff line number Diff line change @@ -49,7 +49,7 @@ The following diagram illustrates the architecture of the **StepMesh** communica
4949
5050- Ubuntu OS
5151- Servers with RDMA NICs and GPU cards.
52- - PyTorch
52+ - torch >= 2.0.0
5353- CUDA NVCC or other compilers
5454
5555## Build
@@ -63,13 +63,13 @@ bash tools/install_deps.sh # only once
6363
6464Build StepMesh
6565``` bash
66- # Please check that your CUDA_HOME is correct
66+ # Please check that your CUDA_HOME is correct and torch is installed.
6767
6868# Build AF library
6969make af
7070
7171# Build and install Fserver (AF's Python SDK)
72- pip3 install -v -e .
72+ pip3 install -v -e . --no-build-isolation
7373```
7474
7575Build without cuda
@@ -80,7 +80,7 @@ Build without cuda
8080USE_CUDA=0 make af
8181
8282# Build and install Fserver (AF's Python SDK)
83- USE_CUDA=0 pip3 install -v -e .
83+ USE_CUDA=0 pip3 install -v -e . --no-build-isolation
8484
8585```
8686
Original file line number Diff line number Diff line change 11[build-system ]
22requires = [
33 " setuptools>=40.8.0" ,
4- " wheel" ,
5- " torch>=2.0.0"
4+ " wheel"
65]
76build-backend = " setuptools.build_meta"
You can’t perform that action at this time.
0 commit comments