Skip to content

Commit

Permalink
Merge pull request #6 from perftool-incubator/no-cuda-userenv
Browse files Browse the repository at this point in the history
We don't need this userenv any more
  • Loading branch information
atheurer authored Aug 1, 2024
2 parents c098829 + 7526b51 commit d7ce43d
Showing 1 changed file with 1 addition and 51 deletions.
52 changes: 1 addition & 51 deletions workshop.json
Original file line number Diff line number Diff line change
Expand Up @@ -8,57 +8,7 @@
{
"name": "rhel-ai",
"requirements": []
},
{
"name": "cuda-rhubi9",
"requirements": [
"python-pip-packages",
"os-pkgs",
"instructlab"
]
}
],
"requirements": [
{
"name": "python-pip-packages",
"type": "python311",
"python3_info": {
"packages": [
"torch==2.3.0 --index-url https://download.pytorch.org/whl/cu121",
"packaging wheel",
"flash-attn==2.5.9.post1",
"deepspeed==0.14.2",
"transformers==4.40.1",
"ipdb",
"jupyterlab",
"gpustat",
"matplotlib",
"hydra-core",
"datasets",
"rich",
"numba",
"'numpy<2.0.0'"
]
}
},
{
"name": "os-pkgs",
"type": "distro",
"distro_info": {
"packages": [
"libnccl",
"libaio-devel"
]
}
},
{ "name": "instructlab",
"type": "manual",
"manual_info": {
"commands": [
"git clone https://github.com/instructlab/training.git /var/run/training",
"pushd /var/run/training; python3.11 -m pip install ."
]
}
}
]
"requirements": []
}

0 comments on commit d7ce43d

Please sign in to comment.