在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. 在容器内部调试代码

相关推荐
一技安身8 小时前
【信创】Docker‑Compose V2 两种离线部署(独立模式、插件模式)简易教程
java·docker·eureka
鸿观工坊12 小时前
Docker 多阶段构建实践指南
docker
大牧师14 小时前
MySQL 学习教程
数据库·sql·mysql·docker·node·全栈·后端数据
一叶龙洲15 小时前
Ubuntu24.04 Cursor导入vscode配置
vscode·编辑器
天外天-亮16 小时前
docker + window 安装
运维·docker·容器
栀椩16 小时前
CODrone 无人机航拍车辆检测
pytorch·python·yolo
进击的小菜鸡dd16 小时前
docker 容器的挂载映射关系查看
运维·docker·容器
盼小辉丶17 小时前
PyTorch计算机视觉(9)——变分自编码器(VAE)详解与实现
人工智能·pytorch·深度学习·计算机视觉
程序员老赵17 小时前
Docker 部署 TeX Live:轻松搭建 LaTeX 论文排版编译平台
前端·docker·latex
clorinda17 小时前
PyTorch 手写数字识别学习笔记
pytorch·笔记·学习