Merge pull request #5 from akay/patch-1

add nvidia-docker to the list of prerequisites
This commit is contained in:
Brendan Dolan-Gavitt 2022-08-03 21:23:33 -04:00 committed by GitHub
commit 958ee6f036
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -9,6 +9,7 @@ You'll need:
* Docker * Docker
* `docker-compose` >= 1.28 * `docker-compose` >= 1.28
* An NVIDIA GPU with enough VRAM to run the model you want. * An NVIDIA GPU with enough VRAM to run the model you want.
* [`nvidia-docker`](https://github.com/NVIDIA/nvidia-docker)
Note that the VRAM requirements listed by `setup.sh` are *total* -- if you have multiple GPUs, you can split the model across them. So, if you have two NVIDIA RTX 3080 GPUs, you *should* be able to run the 6B model by putting half on each GPU. Note that the VRAM requirements listed by `setup.sh` are *total* -- if you have multiple GPUs, you can split the model across them. So, if you have two NVIDIA RTX 3080 GPUs, you *should* be able to run the 6B model by putting half on each GPU.