machine-learning-deep-learning/Machine-Learning-Collection
1
0
Fork 0
mirror of https://github.com/aladdinpersson/Machine-Learning-Collection.git synced 2026-02-21 11:18:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

updated readme for docker

Browse Source
This commit is contained in:
Aladdin Persson
2023-12-08 15:32:35 +01:00
parent 1b310a3422
commit 5ccb93b033
11 changed files with 49 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
Dockerfile
View File

@@ -1,4 +1,5 @@
FROM pytorch/pytorch:latest
FROM pytorch/pytorch:2.1.1-cuda12.1-cudnn8-runtime
RUN pip install --upgrade pip
WORKDIR /code
COPY requirements.txt /code/
RUN pip install --no-cache-dir -r requirements.txt
EXPOSE 8888