@@ -9,52 +9,52 @@ tables by explicitly configuring them.
9
9
Installation
10
10
------------
11
11
12
- Requirements: - python >= 3.7
12
+ Requirements: - Python >= 3.7
13
13
14
14
We highly recommend CUDA when using torchRec. If using CUDA: - cuda >=
15
15
11.0
16
16
17
17
.. code :: python
18
18
# TODO : replace these
19
- # install conda to make installying pytorch with cudatoolkit 11.3 easier.
19
+ # install Conda to make installing PyTorch with cudatoolkit 11.3 easier.
20
20
! sudo rm Miniconda3- py37_4.9.2- Linux- x86_64.sh Miniconda3- py37_4.9.2- Linux- x86_64.sh.*
21
21
! sudo wget https:// repo.anaconda.com/ miniconda/ Miniconda3- py37_4.9.2- Linux- x86_64.sh
22
22
! sudo chmod + x Miniconda3- py37_4.9.2- Linux- x86_64.sh
23
23
! sudo bash ./ Miniconda3- py37_4.9.2- Linux- x86_64.sh - b - f - p / usr/ local
24
24
25
25
.. code :: python
26
26
27
- # install pytorch with cudatoolkit 11.3
27
+ # install PyTorch with cudatoolkit 11.3
28
28
! sudo conda install pytorch cudatoolkit= 11.3 - c pytorch- nightly - y
29
29
30
- Installing torchRec will also install
30
+ Installing TorchRec will also install
31
31
`FBGEMM <https://github.com/pytorch/fbgemm >`__, a collection of CUDA
32
- kernels and GPU enabled operations to run
32
+ kernels and GPU enabled operations to run.
33
33
34
34
.. code :: python
35
35
36
36
# install torchrec
37
37
! pip3 install torchrec- nightly
38
38
39
- Install multiprocess which works with ipython to for multi-processing
40
- programming within colab
39
+ Install ` multiprocess`` which works with ` iPython ` for multi-processing
40
+ programming within `Colab``:
41
41
42
42
.. code :: python
43
43
44
44
! pip3 install multiprocess
45
45
46
46
The following steps are needed for the Colab runtime to detect the added
47
- shared libraries. The runtime searches for shared libraries in /usr/lib,
48
- so we copy over the libraries which were installed in /usr/local/lib/.
49
- **This is a very necessary step, only in the colab runtime **.
47
+ shared libraries. The runtime searches for shared libraries is in ` /usr/lib ` ,
48
+ so we copy over the libraries which were installed in ` /usr/local/lib/ ` .
49
+ **This is a very necessary step, only in the Colab runtime **.
50
50
51
51
.. code :: python
52
52
53
53
! sudo cp / usr/ local/ lib/ lib* / usr/ lib/
54
54
55
55
**Restart your runtime at this point for the newly installed packages
56
56
to be seen. ** Run the step below immediately after restarting so that
57
- python knows where to look for packages. **Always run this step after
57
+ Python knows where to look for packages. **Always run this step after
58
58
restarting the runtime. **
59
59
60
60
.. code :: python
@@ -71,7 +71,7 @@ Due to the notebook enviroment, we cannot run
71
71
can do multiprocessing inside the notebook to mimic the setup. Users
72
72
should be responsible for setting up their own
73
73
`SPMD <https://en.wikipedia.org/wiki/SPMD >`_ launcher when using
74
- Torchrec . We setup our environment so that torch distributed based
74
+ TorchRec . We setup our environment so that torch distributed based
75
75
communication backend can work.
76
76
77
77
.. code :: python
0 commit comments