Mingqian-233 commited on
Commit
8494e73
·
verified ·
1 Parent(s): 8ad8408

Update final docs/environment.md

Browse files
Files changed (1) hide show
  1. docs/environment.md +52 -0
docs/environment.md ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Environment
2
+
3
+ ## Validated Jittor setup
4
+
5
+ | Component | Version |
6
+ |---|---|
7
+ | Python | 3.10 |
8
+ | Jittor | 1.3.8.5 |
9
+ | NumPy | 1.26.4 |
10
+ | CUDA toolchain | Jittor CUDA 11.2 + cuDNN 8 |
11
+ | C++ compiler | g++-10 |
12
+
13
+ Install the Python dependencies and expose the repository package:
14
+
15
+ ```bash
16
+ pip install -r requirements.txt
17
+ export cc_path=/usr/bin/g++-10
18
+ export PYTHONPATH="$PWD:$PWD/python"
19
+ ```
20
+
21
+ Jittor compiles operators on first use, so the first model construction can
22
+ take several minutes.
23
+
24
+ ## Compatibility notes
25
+
26
+ ### NumPy
27
+
28
+ Keep NumPy at 1.26.4. Jittor 1.3.8.5 combined with NumPy 2.x can produce
29
+ incorrect values for operations consuming NumPy-backed arrays without raising
30
+ an exception. Verify the installation with:
31
+
32
+ ```bash
33
+ python - <<'PY'
34
+ import jittor as jt
35
+ x = jt.float32([1, 2, 3])
36
+ assert (x + x).numpy().tolist() == [2.0, 4.0, 6.0]
37
+ print('Jittor array check passed')
38
+ PY
39
+ ```
40
+
41
+ ### Compiler
42
+
43
+ Jittor's CUDA 11.2 frontend is incompatible with newer system compiler headers
44
+ on some Linux distributions. The validated toolchain uses g++-10 selected by
45
+ the lowercase `cc_path` environment variable.
46
+
47
+ ### Reference environment
48
+
49
+ Regenerating PyTorch goldens or using the reference DOTA merge metric requires
50
+ the original Point2RBox-v3 environment: PyTorch 2.2, torchvision 0.17,
51
+ mmengine 0.10.7, mmcv 2.2.0, mmdet 3.3.0 and the reference mmrotate package.
52
+ Regular Jittor training and inference do not require this second environment.