在windows10系统上安装docker,然后在容器中运行GPU版本的Pytorch,并使用vscode连接该容器

一 . 安装Docker Desktop

  1. 首先打开网址https://docs.docker.com/desktop/install/windows-install/

  2. 下载完后,双击下面的exe文件进行安装,默认情况下,Docker Desktop 安装在C:\Program Files\Docker\Docker

  3. 出现提示时,请确保根据您选择的后端,在配置页面上选择使用 WSL 2 而不是 Hyper-V选项。

    如果您的系统仅支持两个选项中的一个,您将无法选择使用哪个后端。

  4. 按照安装向导上的说明授权安装程序并继续安装。

  5. 安装成功后,选择关闭,完成安装过程。

  6. Docker Desktop的界面说明,使用方法,配置镜像源,镜像存放路径修改

    可以参考https://blog.csdn.net/GoodburghCottage/article/details/131413312

  7. 安装完成Docker Desktop后,首先打开Docker Desktop,然后就可以在终端执行docker相关命令(此时就不要再使用命令systemctl start docker来启动Docker,因为它默认已经自动启动),如下所示:

二. 在容器中运行 GPU版本的pytorch

这里共有两种方法,其中方法一是直接拉取一个pytorch镜像,然后运行这个镜像即可;方法二是拉取一个带有cuda和cudnn的ubuntu系统镜像,然后运行该镜像并在里面安装pytorch。

显然,第一种方法更简单。

方法一

  1. 打开网址:https://catalog.ngc.nvidia.com/containers,网页界面如下所示:
  2. 查看不同版本的pytorch镜像所包括的信息

注意:pytorch镜像的版本和pytorch版本并不等同,不同版本的pytorch镜像中所安装的pytorch版本可能是不同的




  1. 查看不同版本的Pytorch镜像
  2. 这里以22.03版本的pytorch为例,进行镜像的拉取和运行操作

方法二

可以参考https://blog.csdn.net/qq_41204464/article/details/132891018

三. 使用vscode连接容器

  1. 首先安装docker插件

    安装完该插件后,在vscode左侧会显示docker图标,并且也会展示本地的所有镜像和容器此时就可以在通过vscode终端连接到容器内部,具体操作如下:

注意:此时虽然可以连接容器,但是调试容器内部的代码非常不方便,所以还需要安装另外一个插件

  1. 安装Dev Containers插件

  2. 进入docker中的容器内部



  3. 在容器内部运行代码

  4. 在容器内部调试代码

相关推荐
码间留白5 小时前
手把手把 GitHub 推上阿里云 ECS:Actions CI/CD 全链路避坑实录(折腾好几天,到 2 分钟见绿)
运维·阿里云·ci/cd·docker·devops·github actions
Vex2une5 小时前
Docker Desktop WSL2后端:2026年性能调优与避坑实战指南
windows·docker·desktop·gpu容器·容器开发·virtiofs·buildkit
程序员老赵6 小时前
Docker 部署青龙面板:轻松搭建定时任务管理平台
前端·docker·开源
TTATTC6 小时前
为什么单会话聊天做不好复杂工程?Claude Code 创作者的并发 Worktree 与记忆闭环实践
docker·开源
TTATTC7 小时前
状态机驱动长事务,还是 DAG 编排高并发?高可靠业务流程引擎的双核架构设计
docker·开源
殷忆枫7 小时前
告别依赖地狱!在 Ubuntu 上用 Docker 极速部署 Home Assistant 与 MQTT,并接入 Siri 与嵌入式 C 终端
linux·ubuntu·docker
田里的水稻8 小时前
PyTorch_Dataloader\TensorBoard---数据加载与可视化
人工智能·pytorch·python·神经网络·机器学习
田里的水稻8 小时前
FA_CNN卷积神经网络三---非线性激活函数PyTorch
人工智能·pytorch·神经网络·机器学习·cnn
千桐科技8 小时前
qModel算法模型平台功能解析:从源码到 Docker 镜像,部署能自动完成哪些工作?
运维·docker·容器·qmodel
江湖有缘11 小时前
Docker实战 | 使用Docker部署Imagor Studio图片编辑工具
java·docker·eureka