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。

相关推荐
螺蛳粉 螺蛳粉23 分钟前
第 02 篇:镜像分层与 containerd 2.x 实战:SystemdCgroup 与 pause 镜像配置踩坑(底座认知 02)
docker·云原生·容器·kubernetes·etcd·containerd·kube-apiserver
wefg11 小时前
【Docker】镜像相关指令
运维·docker·容器
java_logo2 小时前
Docker 部署 DPanel:轻松搭建轻量容器管理平台
运维·docker·容器·podman·容器管理·dpanel·轩辕镜像
死磕音视频2 小时前
Open WebUI 入门:用 Docker 装上本地对话界面并连接 Ollama
docker·ollama·本地大模型·open webui
分布式存储与RustFS2 小时前
CI 老被 Docker Hub 的 429 卡住,自建 Registry 把镜像搬进内网
运维·docker·开源·镜像仓库·对象存储·ci·rustfs
田里的水稻3 小时前
FA_CNN卷积神经网络十八---感知机到多层感知机PyTorch
人工智能·pytorch·python
n112123 小时前
Docker 容器时区错 8 小时:3 个坑
运维·docker·容器·故障排查·服务器运维·容器时区
听风吹等浪起4 小时前
第27章:EfficientNet全家桶|钢材表面6类缺陷分类完整Pytorch实现
pytorch·分类·数据挖掘
Eiceblue4 小时前
用 Python 给 Word 文档加水印:文字、图片与批量处理
开发语言·vscode·python·word
慈云数据4 小时前
用 DSH 辅助部署一个网站监控平台:Uptime Kuma + Docker 实战
运维·docker·容器