大连密封材料 发表于 2025-3-19 00:09:26

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

一 . 安装Docker Desktop


[*] 首先打开网址https://docs.docker.com/desktop/install/windows-install/
https://i-blog.csdnimg.cn/direct/ccc6ddaa53ea47f98c594f71abd86e09.png
[*] 下载完后,双击下面的exe文件举行安装,默认情况下,Docker Desktop 安装在C:\Program Files\Docker\Docker
https://i-blog.csdnimg.cn/direct/422e51e48a38461dbaefed28f2cf2446.png
[*] 出现提示时,请确保根据您选择的后端,在设置页面上选择使用 WSL 2 而不是 Hyper-V选项。
假如您的系统仅支持两个选项中的一个,您将无法选择使用哪个后端。
[*] 按照安装向导上的阐明授权安装程序并继承安装。
[*] 安装成功后,选择关闭,完成安装过程。
[*] Docker Desktop的界面阐明,使用方法,设置镜像源,镜像存放路径修改
可以参考https://blog.csdn.net/GoodburghCottage/article/details/131413312
[*] 安装完成Docker Desktop后,首先打开Docker Desktop,然后就可以在终端执行docker相干下令(此时就不要再使用下令systemctl start docker来启动Docker,因为它默认已经自动启动),如下所示:
https://i-blog.csdnimg.cn/direct/c047bf2ebc934cb198b725ae9fced8b7.png
二. 在容器中运行 GPU版本的pytorch

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


[*]打开网址:https://catalog.ngc.nvidia.com/containers,网页界面如下所示:
https://i-blog.csdnimg.cn/direct/b9688b3206074a2e8e641f8a80b6bf85.png
[*]查看不同版本的pytorch镜像所包括的信息
   留意:pytorch镜像的版本和pytorch版本并不等同,不同版本的pytorch镜像中所安装的pytorch版本可能是不同的
https://i-blog.csdnimg.cn/direct/3ed26d9490db4945b9a7f565c9736885.png
https://i-blog.csdnimg.cn/direct/66eb490ddff3415da2679b6a55a9cdf7.png
https://i-blog.csdnimg.cn/direct/80938db1f03648f19b7ddfc91ca66b9e.png
https://i-blog.csdnimg.cn/direct/1e36ae712ecd40a187248f24508c7433.png
https://i-blog.csdnimg.cn/direct/96820856c0be479c9c2f3b2a859df8d7.png

[*]查看不同版本的Pytorch镜像
https://i-blog.csdnimg.cn/direct/79dfe06afef942b6b608b9b3dff0cfd6.png
[*]这里以22.03版本的pytorch为例,举行镜像的拉取和运行操作
https://i-blog.csdnimg.cn/direct/2f3c1fdcc9854707a5a4b95bcee56113.png
https://i-blog.csdnimg.cn/direct/23dfad19988d4f84bcf7175d77597c9f.png
方法二

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


[*]首先安装docker插件
https://i-blog.csdnimg.cn/direct/fba12d8c53dc44e6b31571e08ab54111.png
安装完该插件后,在vscode左侧会表现docker图标,并且也会展示当地的全部镜像和容器https://i-blog.csdnimg.cn/direct/1ae4796c58654b32bca0064db6839660.png​此时就可以在通过vscode终端毗连到容器内部,详细操作如下:
   留意:此时虽然可以毗连容器,但是调试容器内部的代码非常不方便,所以还需要安装另外一个插件
https://i-blog.csdnimg.cn/direct/fbeece910d8b4bedac4a5ff6c7d18789.png
2. 安装Dev Containers插件
https://i-blog.csdnimg.cn/direct/57bd6d91d24c49858ad352e45a828101.png
3. 进入docker中的容器内部
https://i-blog.csdnimg.cn/direct/1a5d89065961443f9529f4389f47f90e.png
https://i-blog.csdnimg.cn/direct/e16d3f868bae403b9b98420f611324fb.png
https://i-blog.csdnimg.cn/direct/dc4f602d777842a7b2ebcc71dc17f7c6.png
https://i-blog.csdnimg.cn/direct/be6801e0b20545699bc1e14e02698d31.png

[*]在容器内部运行代码
https://i-blog.csdnimg.cn/direct/0237ae9a04e14796a57f43cd8a681a92.png
[*]在容器内部调试代码
https://i-blog.csdnimg.cn/direct/14d6e2f3d47d4b41b7275c978bcf9099.png

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: 在windows10系统上安装docker,然后在容器中运行GPU版本的Pytorch,并使用vscode毗连该容器