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