diff --git a/docker-compose.yml b/docker-compose.yml index 4a48e0c..92940c7 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -11,7 +11,7 @@ services: - infinity volumes: - /usr/lib/x86_64-linux-gnu/nvidia/current:/cuda:ro - - $PWD/folder:/folder + - ./folder:/folder environment: - NVIDIA_VISIBLE_DEVICES=all devices: