pin Hindsight retrieval models to GPU0
This commit is contained in:
@@ -11,6 +11,7 @@ services:
|
|||||||
- float16
|
- float16
|
||||||
environment:
|
environment:
|
||||||
HF_HUB_CACHE: /data
|
HF_HUB_CACHE: /data
|
||||||
|
NVIDIA_VISIBLE_DEVICES: "0"
|
||||||
HTTP_PROXY: http://192.168.50.45:7890
|
HTTP_PROXY: http://192.168.50.45:7890
|
||||||
HTTPS_PROXY: http://192.168.50.45:7890
|
HTTPS_PROXY: http://192.168.50.45:7890
|
||||||
NO_PROXY: 127.0.0.1,localhost,192.168.50.0/24
|
NO_PROXY: 127.0.0.1,localhost,192.168.50.0/24
|
||||||
@@ -49,6 +50,7 @@ services:
|
|||||||
- float16
|
- float16
|
||||||
environment:
|
environment:
|
||||||
HF_HUB_CACHE: /data
|
HF_HUB_CACHE: /data
|
||||||
|
NVIDIA_VISIBLE_DEVICES: "0"
|
||||||
HTTP_PROXY: http://192.168.50.45:7890
|
HTTP_PROXY: http://192.168.50.45:7890
|
||||||
HTTPS_PROXY: http://192.168.50.45:7890
|
HTTPS_PROXY: http://192.168.50.45:7890
|
||||||
NO_PROXY: 127.0.0.1,localhost,192.168.50.0/24
|
NO_PROXY: 127.0.0.1,localhost,192.168.50.0/24
|
||||||
|
|||||||
Reference in New Issue
Block a user