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

打印 上一主题 下一主题

主题 913|帖子 913|积分 2739

一 . 安装Docker Desktop


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

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

  • 出现提示时,请确保根据您选择的后端,在设置页面上选择使用 WSL 2 而不是 Hyper-V选项。
    假如您的系统仅支持两个选项中的一个,您将无法选择使用哪个后端。
  • 按照安装向导上的阐明授权安装程序并继承安装。
  • 安装成功后,选择关闭,完成安装过程。
  • Docker Desktop的界面阐明,使用方法,设置镜像源,镜像存放路径修改
    可以参考https://blog.csdn.net/GoodburghCottage/article/details/131413312
  • 安装完成Docker Desktop后,首先打开Docker Desktop,然后就可以在终端执行docker相干下令(此时就不要再使用下令systemctl start docker来启动Docker,因为它默认已经自动启动),如下所示:

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

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


  • 打开网址:https://catalog.ngc.nvidia.com/containers,网页界面如下所示:

  • 查看不同版本的pytorch镜像所包括的信息
   留意:pytorch镜像的版本和pytorch版本并不等同,不同版本的pytorch镜像中所安装的pytorch版本可能是不同的





  

  • 查看不同版本的Pytorch镜像

  • 这里以22.03版本的pytorch为例,举行镜像的拉取和运行操作


方法二

可以参考https://blog.csdn.net/qq_41204464/article/details/132891018
三. 使用vscode毗连容器


  • 首先安装docker插件

    安装完该插件后,在vscode左侧会表现docker图标,并且也会展示当地的全部镜像和容器
    ​此时就可以在通过vscode终端毗连到容器内部,详细操作如下:
   留意:此时虽然可以毗连容器,但是调试容器内部的代码非常不方便,所以还需要安装另外一个插件
  

2. 安装Dev Containers插件

3. 进入docker中的容器内部





  • 在容器内部运行代码

  • 在容器内部调试代码


免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

大连密封材料

金牌会员
这个人很懒什么都没写!
快速回复 返回顶部 返回列表