Windows环境下Pytorch项目搭建在Docker中运行
1. 安装windows版本的Docker
网上已有诸多博客教程,这里就不再赘述。
2. 搭建本地Pytorch环境
搭建本地Pytorch环境的方式我使用了两种方式,推荐使用第一种。
第一种:
(1)在docker hub中(https://hub.docker.com),找到自己版本的pytorch版本,我这里是使用的torch==
)
(2)根据自己所需的pytorch版本,将镜像拉入本地
docker pull anibali/pytorch:1.10.2-cuda11.3
)
(3)docker images查看自己本地镜像,检查pytorch镜像是否已经拉入本地
(4)根据镜像id运行镜像,并进入镜像生成的容器
docker run -it 镜像id号 bash
(5)查看容器中安装的包
pip list
(6)如果缺少自己所需要的包,可以在容器中安装
例如:pip install gym
(7)新建src文件夹和Dockerfile文件,将本地pycharm的pytorch代码复制到自己创建src文件夹中
(8)Dockerfile文件中的内容为:
FROM anibali/pytorch:1.10.2-cuda11.3
# WORKDIR 根据自己创建的Dockerfile文件所在路径填写
WORKDIR D:\study\docker\fl-pro
ADD . .
# ./src/t.py 该文件是docker容器启动运行的文件
CMD ["python", "./src/t.py"]
(9)将本地项目打包成docker镜像
docker build -t fl:0.1 .
标签:Windows,pytorch,Pytorch,本地,镜像,docker,Docker From: https://www.cnblogs.com/miaoyindian/p/18054176(10)打包成功,可在镜像中查看