reverting changes to docker-compose.yml
@@ -193,7 +193,7 @@ services:
context: ./
dockerfile: dp/dockerfile_skill_gpu
environment:
- - CUDA_VISIBLE_DEVICES=2
+ - CUDA_VISIBLE_DEVICES=0
deploy:
mode: replicated
replicas: 1
@@ -618,7 +618,7 @@ services:
context: ./
dockerfile: dp/dockerfile_skill_gpu
environment:
- - CUDA_VISIBLE_DEVICES=2
+ CUDA_VISIBLE_DEVICES: ""
deploy:
mode: replicated
replicas: 1
@@ -648,7 +648,7 @@ services:
command: gunicorn --workers=1 server:app -b 0.0.0.0:8053
volumes: []
environment:
- - CUDA_VISIBLE_DEVICES=1
+ DEVICE: gpu_0
deploy:
mode: replicated
replicas: 1
@@ -668,7 +668,7 @@ services:
command: gunicorn --workers=1 server:app -b 0.0.0.0:8065
volumes: []
environment:
- - CUDA_VISIBLE_DEVICES=1
+ DEVICE: gpu_0
deploy:
mode: replicated
replicas: 1
@@ -706,7 +706,7 @@ services:
context: ./
dockerfile: dp/dockerfile_skill_gpu
environment:
- - CUDA_VISIBLE_DEVICES=1
+ CUDA_VISIBLE_DEVICES: "0"
deploy:
mode: replicated
replicas: 1
