/oneflow-dev-dockerfile

dockerfile for oneflow building env

Primary LanguageShell

DockerFiles

include

  • git
  • tmux
  • sshd
  • neovimplus
  • zsh & oh-my-zsh
    • zsh-autosuggestions
    • zsh-syntax-highlighting
    • z
    • extractor
  • pyenv & python3.8.5

Build

docker build --build-arg USER_UID=$(id -u) -t cuda102:1.2 -f ./Dockerfile-cuda102 .
docker build --build-arg USER_UID=$(id -u) -t cuda112:1.0 -f ./Dockerfile-cuda112 .

如果到apt-get update卡住，可能是网络问题，可以使用host模式构建:

docker build --build-arg USER_UID=$(id -u) --network host -t cuda102:1.2 -f ./Dockerfile-cuda102 .
docker build --build-arg USER_UID=$(id -u) --network host -t cuda112:1.0 -f ./Dockerfile-cuda112 .

Run

docker run -it --shm-size=8G -P --privileged --runtime=nvidia -u YOUR_UID --name "NAME" -v PATH_LOCAL:PATH_DOCKER DOCKER_IMAGE /bin/zsh