Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

WIP test out new dockerfile with more nvidia tools #1557

Open
wants to merge 14 commits into
base: main
Choose a base branch
from
Open

WIP test out new dockerfile with more nvidia tools #1557

Changes from 1 commit
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
7290e77
WIP test out new dockerfile with more nvidia tools
winglian Apr 21, 2024
2ca30d1
make sure to install axolotl :facepalm:
winglian Apr 21, 2024
89086d3
install venv from apt
winglian Apr 21, 2024
f076933
add -yes option to pip install
winglian Apr 21, 2024
493b6b9
install git-lfs
winglian Apr 21, 2024
dd87c81
make sure to install packaing too
winglian Apr 21, 2024
b9211fc
use --system-site-packages to pickup global packaging pkg
winglian Apr 21, 2024
b8aba89
setup /workspace
winglian Apr 21, 2024
4be78e9
no need for mkdir
winglian Apr 21, 2024
b4df593
make sure to cd to axolotl dir
winglian Apr 22, 2024
b5e2821
reorder for fix
winglian Apr 22, 2024
f807516
use torch 2.2.2
winglian Apr 22, 2024
8401b6f
shuffle some dependency versions around
winglian Apr 22, 2024
95c2e5f
agree to uninstall
winglian Apr 22, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
install venv from apt
  • Loading branch information
@winglian
winglian committed Apr 21, 2024
commit 89086d3d1c67ba01eb17fad9fc4a6c8881f771ab
2 changes: 2 additions & 0 deletions docker/Dockerfile-beta
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
FROM nvcr.io/nvidia/pytorch:24.03-py3

RUN apt update && apt install python3.10-venv

RUN python3 -m pip install --upgrade pip

RUN groupadd axolotl && useradd -m -g axolotl -s /bin/bash axolotl
Expand Down
Loading