diff --git a/.nojekyll b/.nojekyll index c91edf24c..f4df579b9 100644 --- a/.nojekyll +++ b/.nojekyll @@ -1 +1 @@ -45aa1b5b \ No newline at end of file +f1264b57 \ No newline at end of file diff --git a/docs/docker.html b/docs/docker.html index d581217d1..fc8f7438f 100644 --- a/docs/docker.html +++ b/docs/docker.html @@ -494,6 +494,19 @@ pre > code.sourceCode > span > a:first-child::before { text-decoration: underlin
This section describes the different Docker images that are released by AxolotlAI at Docker Hub.
+For Blackwell GPUs, please use the tags with Pytorch 2.7.0 and CUDA 12.8.
+The base image is the most minimal image that can install Axolotl. It is based on the nvidia/cuda image. It includes python, torch, git, git-lfs, awscli, pydantic, and more.
Follow the instructions at: https://pytorch.org/get-started/locally/
+For Blackwell GPUs, please use Pytorch 2.7.0 and CUDA 12.8.
+pip3 install -U packaging setuptools wheel ninja
@@ -558,6 +571,19 @@ Advanced Docker Configuration
axolotlai/axolotl:main-latestFor Blackwell GPUs, please use axolotlai/axolotl:main-py3.11-cu128-2.7.0 or the cloud variant axolotlai/axolotl-cloud:main-py3.11-cu128-2.7.0.
Please refer to the Docker documentation for more information on the different Docker images that are available.