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。

相关推荐
lcx_defender11 分钟前
【Docker】Docker部署运行nacos
运维·docker·容器
啦啦啦小石头43 分钟前
docker添加用户权限不使用sudo
运维·docker·容器
cuber膜拜1 小时前
Weaviate 简介与基本使用
数据库·python·docker·向量数据库·weaviate
xuefuhe1 小时前
Docker常用命令
docker
取个鸣字真的难2 小时前
简单快速的用 Claude Code 帮你创建 PPT 生成 Skills
vscode·powerpoint·ai编程
云草桑2 小时前
15分钟快速了解 Odoo
数据库·python·docker·postgresql·.net·odoo
春日见2 小时前
Autoware使用教程
大数据·人工智能·深度学习·elasticsearch·搜索引擎·docker·容器
阿_焦2 小时前
docker部署dify踩坑笔记
docker·ai·dify
lcx_defender2 小时前
【Docker】Docker部署运行Kibana
运维·docker·容器
大模型玩家七七3 小时前
证据不足 vs 证据冲突:哪个对模型更致命
数据库·人工智能·pytorch·深度学习·安全