feat: add Jupyter notebook for linear regression and CIFAR-10 image classification with model training
This commit is contained in:
@ -4,13 +4,13 @@ FROM pytorch/pytorch:latest
|
||||
# 安装 Jupyter Lab 和常用库
|
||||
RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple && \
|
||||
pip install --upgrade pip && \
|
||||
pip install jupyterlab matplotlib pandas seaborn
|
||||
pip install jupyterlab matplotlib pandas seaborn tensorboard visdom
|
||||
|
||||
# 设置默认工作目录
|
||||
WORKDIR /workspace
|
||||
|
||||
# 暴露 Jupyter 端口
|
||||
EXPOSE 8888
|
||||
# 暴露端口:Jupyter、TensorBoard、Visdom
|
||||
EXPOSE 8888 6006 8097
|
||||
|
||||
# 默认启动 Jupyter Lab
|
||||
CMD ["jupyter", "lab", "--ip=0.0.0.0", "--port=8888", "--no-browser", "--allow-root"]
|
||||
|
||||
BIN
data/images/dog.jpg
Normal file
BIN
data/images/dog.jpg
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 137 KiB |
Reference in New Issue
Block a user