fada/pytorch-study
1
0
Fork 0
Code Issues Wiki Activity

feat: add Jupyter notebook for linear regression and CIFAR-10 image classification with model training

Browse Source
This commit is contained in:
fada
2025-06-20 16:16:00 +08:00
parent 30607ae7c1
commit d1e04739d9
3 changed files with 1549 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1546
14.ipynb Normal file
View File

File diff suppressed because one or more lines are too long

6
Dockerfile
View File

@ -4,13 +4,13 @@ FROM pytorch/pytorch:latest
# 安装 Jupyter Lab 和常用库 # 安装 Jupyter Lab 和常用库
RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \ RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
pip install --upgrade pip && \ pip install --upgrade pip && \
pip install jupyterlab matplotlib pandas seaborn pip install jupyterlab matplotlib pandas seaborn tensorboard visdom
# 设置默认工作目录 # 设置默认工作目录
WORKDIR /workspace WORKDIR /workspace
# 暴露 Jupyter 端口 # 暴露端口：Jupyter、TensorBoard、Visdom
EXPOSE 8888 EXPOSE 8888 6006 8097
# 默认启动 Jupyter Lab # 默认启动 Jupyter Lab
CMD ["jupyter", "lab", "--ip=0.0.0.0", "--port=8888", "--no-browser", "--allow-root"] CMD ["jupyter", "lab", "--ip=0.0.0.0", "--port=8888", "--no-browser", "--allow-root"]

BIN
data/images/dog.jpg Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 137 KiB