Add Dockerfile.gpu for GPU support

This commit is contained in:
Aurélien Geron
2021-03-03 22:19:27 +13:00
parent abd75d7fd7
commit d6c2ba1b8f
4 changed files with 265 additions and 4 deletions

View File

@@ -3,7 +3,7 @@ services:
handson-ml2:
build:
context: ../
dockerfile: ./docker/Dockerfile
dockerfile: ./docker/Dockerfile #Dockerfile.gpu
args:
- username=devel
- userid=1000
@@ -20,3 +20,8 @@ services:
volumes:
- ../:/home/devel/handson-ml2
command: /opt/conda/envs/tf2/bin/jupyter notebook --ip='0.0.0.0' --port=8888 --no-browser
#deploy:
# resources:
# reservations:
# devices:
# - capabilities: [gpu]