books/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2026-04-10 12:33:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

updates for PyTorch 2.5 (#408)

Browse Source 
* updated Dockerfile

* updated MHA implementations for PT 2.5

* fixed typo

* update installation instruction

* Update setup/03_optional-docker-environment/.devcontainer/Dockerfile

---------

Co-authored-by: rasbt <mail@sebastianraschka.com>
This commit is contained in:
Daniel Kleine
2024-10-23 03:23:31 +02:00
committed by GitHub
parent 534a704364
commit ef4018181e
2 changed files with 32 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
setup/03_optional-docker-environment/.devcontainer/Dockerfile
View File

@@ -1,5 +1,7 @@
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
# Install PyTorch 2.5 with CUDA 12.4
FROM pytorch/pytorch:2.5.0-cuda12.4-cudnn9-runtime
# Install Ubuntu packages
RUN apt-get update && \
apt-get upgrade -y && \
apt-get install -y rsync && \
@@ -7,6 +9,7 @@ RUN apt-get update && \
apt-get install -y curl && \
rm -rf /var/lib/apt/lists/*
# Install Python packages
COPY requirements.txt requirements.txt
RUN pip install --upgrade pip
RUN pip install --no-cache-dir -r requirements.txt