lingyuzeng/cdc_dockerfile
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add root_passwd

Browse Source
This commit is contained in:
hotwa
2024-06-20 17:35:51 +08:00
parent 4572f2cb1c
commit 4e917ffc32
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
finetune/docker-compose_pytorch1.13.yml
View File

@@ -17,6 +17,7 @@ services:
CUDA: cu117 CUDA: cu117
CUDA_ARCH_LIST: "80;86" # for RTX 4090, all : "80;86;89;90" CUDA_ARCH_LIST: "80;86" # for RTX 4090, all : "80;86;89;90"
SETUPTOOLS_VERSION: "69.5.1" SETUPTOOLS_VERSION: "69.5.1"
ROOT_PASSWD: "root"
volumes: volumes:
- ./src:/bbtft - ./src:/bbtft
container_name: ubuntu-finetune container_name: ubuntu-finetune
@@ -27,6 +28,7 @@ services:
shm_size: '32gb' shm_size: '32gb'
ports: ports:
- 3227:2222 - 3227:2222
command: ["/usr/sbin/sshd", "-D"]
environment: environment:
- NVIDIA_VISIBLE_DEVICES=all - NVIDIA_VISIBLE_DEVICES=all
- NVIDIA_DRIVER_CAPABILITIES=compute,utility - NVIDIA_DRIVER_CAPABILITIES=compute,utility