Skip to content

Commit

Permalink
Updated README
Browse files Browse the repository at this point in the history
  • Loading branch information
maouw committed Oct 16, 2023
1 parent 298f7fc commit efa55b6
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -179,7 +179,7 @@ If you have problems connecting to the node, it's possible that you haven't set
Once you are connected to the node, you can verify that the SLEAP container has access to the GPUs by running the following command:

```bash
apptainer run --nv docker://ghcr.io/maouw/sleap-container:latest python -c "import sleap; sleap.system_summary()"
apptainer run --nv oras://ghcr.io/maouw/sleap-container:latest python -c "import sleap; sleap.system_summary()"
```

You should get output that looks something like this:
Expand Down Expand Up @@ -216,7 +216,7 @@ cd ~/training_job
The next step is to launch the container:

```bash
apptainer run --nv docker://ghcr.io/maouw/sleap-container:latest bash train-script.sh
apptainer run --nv oras://ghcr.io/maouw/sleap-container:latest bash train-script.sh
```

Apptainer will download the container image from GitHub and launch it on the node. The option `--nv` enables Nvidia GPU support. Once the container has launched, it will instruct `bash` to run the script `train-script.sh`. This script will start the training job.
Expand Down

0 comments on commit efa55b6

Please sign in to comment.