Win11+docker+gpu+vscode+pytorch配置

近期在研究异常检测相关算法,奈何各个经典算法的代码都是基于不同的pytorch版本的,而且好多代码还需要配置knn_cuda和pointnet2_ops_lib,这两个与cuda版本需要满足对应关系,因此通过conda虚拟环境已经无法满足需求了,不得已,只能来试试docker了。

我的具体情况如下:

  1. 系统为Windows11。因为还有一些日常办公需求,以及一些其他开发工作需要在windows上完成,完全使用Ubuntu不太方便;

  2. 需要使用GPU。这个就不需要多解释了,做深度学习必备。

  3. 作为经典算法的集合,anomalib是我最需要的,因此,我决定从配置anomalib开始。

具体步骤如下:

  1. 参考这篇博客,在win11上配置WSL2,并安装docker-desktop,配置GPU等。文章中提到他出现的那两处报错我并没有遇到,所有步骤都是顺利安装的。Windows 下让 Docker Desktop 关联上 NVidia GPU_windows docker nvidia-CSDN博客

  2. 参考这篇博客,配置VsCode。这里需要注意的是,我下载的docker镜像是pytorch1.12.1的官方镜像,已经发布2年多了,比较老。一开始连接的时候,提示"你已经连接到不受VS CODE支持的版本"。这里安装网上查到的方法,把VSCode的版本降低到1.85就可以了。Windows + docker + python + vscode : 使用容器docker搭建python开发环境,无需本地安装python开发组件-CSDN博客 配置完成后,进入到docker环境中,测试torch.cuda.is_available()显示为True。

相关推荐
数据智能老司机12 小时前
PyTorch 深度学习——使用神经网络来拟合数据
pytorch·深度学习
数据智能老司机12 小时前
PyTorch 深度学习——用于图像的扩散模型
pytorch·深度学习
数据智能老司机12 小时前
PyTorch 深度学习——Transformer 是如何工作的
pytorch·深度学习
数据智能老司机1 天前
PyTorch 深度学习——使用张量表示真实世界数据
pytorch·深度学习
数据智能老司机1 天前
PyTorch 深度学习——它始于一个张量
pytorch·深度学习
Sheffield2 天前
Docker的跨主机服务与其对应的优缺点
linux·网络协议·docker
Sheffield2 天前
Alpine是什么,为什么是Docker首选?
linux·docker·容器
马艳泽3 天前
win10下运行Start Broker and Proxy报错解决
docker
Narrastory3 天前
明日香 - Pytorch 快速入门保姆级教程(三)
pytorch·深度学习
用户13573999256603 天前
Windows 从 0 搭建 WSL2 原生 AI 开发环境:Codex + Docker + VSCode
docker