File tree Expand file tree Collapse file tree 2 files changed +17
-9
lines changed Expand file tree Collapse file tree 2 files changed +17
-9
lines changed Original file line number Diff line number Diff line change @@ -97,16 +97,7 @@ test="true" docker-compose up --build
97
97
```
98
98
99
99
If you need to change some application settings, you may update ` config.py ` according to your needs and re-build the image.
100
- For example, you can run dedoc on CUDA by setting ` on_gpu ` to ` True ` in ` config.py ` .
101
100
102
- To run Dedoc on CUDA with Docker use files from ` docker_gpu ` directory
103
- ([ CUDA Container Toolkit] ( https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html )
104
- should be installed on your machine):
105
-
106
- ``` shell
107
- cd docker_gpu
108
- docker-compose up --build
109
- ```
110
101
111
102
## Install dedoc using pip
112
103
Original file line number Diff line number Diff line change
1
+ To run Dedoc on CUDA with Docker use files from ` docker_gpu ` directory
2
+ ([ CUDA Container Toolkit] ( https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html )
3
+ should be installed on your machine):
4
+
5
+ 1 . Set ` on_gpu ` to ` True ` in ` config.py `
6
+ 2 . Run application:
7
+ ``` shell
8
+ cd docker_gpu
9
+ docker-compose up --build
10
+ ```
11
+
12
+ You can change index if CUDA device at ` docker-compose.yml ` :
13
+ ```
14
+ NVIDIA_VISIBLE_DEVICES: 0
15
+ NVIDIA_VISIBLE_DEVICES: 0, 3
16
+ NVIDIA_VISIBLE_DEVICES: all
17
+ ```
You can’t perform that action at this time.
0 commit comments