0

0

VSCode for AI/ML:PyTorch与TensorFlow开发配置

P粉986688829

P粉986688829

发布时间:2026-01-05 14:00:52

|

273人浏览过

|

来源于php中文网

原创

需配置Python虚拟环境、VSCode解释器路径、必要扩展、框架专属调试及GPU依赖;具体包括用pyenv管理Python版本,为PyTorch/TensorFlow分别创建虚拟环境,VSCode中绑定对应解释器,安装Python/Pylance/Jupyter/TensorBoard扩展,定制launch.json调试配置,并按官网指引安装匹配CUDA或Metal的框架GPU版本。

vscode for ai/ml:pytorch与tensorflow开发配置

如果您希望在VSCode中高效进行PyTorch与TensorFlow的AI/ML开发,需完成Python环境、扩展插件、调试配置及框架依赖的协同设置。以下是实现该目标的具体步骤:

本文运行环境:MacBook Pro,macOS Sequoia。

一、安装Python环境与虚拟环境管理

使用独立虚拟环境可避免PyTorch与TensorFlow因CUDA版本、Python兼容性或依赖冲突导致的运行异常,确保各项目环境隔离且可复现。

1、通过Homebrew安装pyenv与python-build插件。

2、使用pyenv install 3.10.14命令安装Python 3.10.14版本。

3、执行pyenv global 3.10.14设为全局默认版本。

4、运行python -m venv ~/venvs/ml-pytorch创建PyTorch专用虚拟环境。

5、运行python -m venv ~/venvs/ml-tf创建TensorFlow专用虚拟环境。

二、配置VSCode Python扩展与解释器路径

VSCode需识别并绑定正确的Python解释器,才能提供语法高亮、智能补全、断点调试等核心功能;不同框架应分别绑定对应虚拟环境中的Python可执行文件。

1、在VSCode中打开命令面板(Cmd+Shift+P)并输入Python: Select Interpreter。

2、点击“Enter interpreter path...”,浏览至~/venvs/ml-pytorch/bin/python。

3、对TensorFlow项目,重复步骤1后选择~/venvs/ml-tf/bin/python。

4、确认状态栏右下角显示已选解释器路径,且无红色波浪线报错。

三、安装必需扩展与Jupyter支持

Python扩展提供基础语言服务,Pylance增强类型推导,Jupyter扩展启用交互式笔记本开发,而TensorBoard扩展可直接在VSCode内可视化训练过程。

1、在扩展市场搜索并安装Python(by Microsoft)。

TTSMaker
TTSMaker

TTSMaker是一个免费的文本转语音工具,提供语音生成服务,支持多种语言。

下载

2、安装Pylance(by Microsoft),确保settings.json中"python.languageServer": "Pylance"已启用。

3、安装Jupyter(by Microsoft),重启VSCode后可新建.ipynb文件并运行代码单元。

4、安装TensorBoard(by ms-python),在终端中启动tensorboard --logdir=runs后,点击右上角“TensorBoard”按钮即可内嵌查看。

四、配置launch.json实现框架级调试

为支持PyTorch分布式训练或TensorFlow Estimator模式下的断点调试,需为每个框架定制launch.json配置,明确指定模块入口、参数传递方式与环境变量注入机制。

1、在项目根目录创建.vscode/launch.json文件。

2、为PyTorch添加配置:设置"module": "torch.distributed.run","args": ["--nproc_per_node=2", "train.py"]。

3、为TensorFlow添加配置:设置"module": "tensorflow.python.keras.applications"不适用,改用"python": "${file}"并前置设置"env": {"TF_CPP_MIN_LOG_LEVEL": "2"}。

4、保存后,在train.py文件中设置断点,按F5启动调试会话。

五、安装GPU加速依赖与验证脚本

本地GPU支持需匹配CUDA Toolkit、cuDNN与框架预编译版本;仅安装PyPI包不足以启用GPU,必须通过官方渠道获取对应CUDA版本的wheel包。

1、访问PyTorch官网,选择CUDA 12.1选项,复制pip install命令并执行。

2、访问TensorFlow官网,确认macOS不支持CUDA,改用tensorflow-macos与tensorflow-metal插件组合。

3、在PyTorch虚拟环境中运行python -c "import torch; print(torch.cuda.is_available())",输出True表示GPU就绪。

4、在TensorFlow虚拟环境中运行python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))",返回非空列表即表示Metal加速生效。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
什么是分布式
什么是分布式

分布式是一种计算和数据处理的方式,将计算任务或数据分散到多个计算机或节点中进行处理。本专题为大家提供分布式相关的文章、下载、课程内容,供大家免费下载体验。

407

2023.08.11

分布式和微服务的区别
分布式和微服务的区别

分布式和微服务的区别在定义和概念、设计思想、粒度和复杂性、服务边界和自治性、技术栈和部署方式等。本专题为大家提供分布式和微服务相关的文章、下载、课程内容,供大家免费下载体验。

251

2023.10.07

json数据格式
json数据格式

JSON是一种轻量级的数据交换格式。本专题为大家带来json数据格式相关文章,帮助大家解决问题。

457

2023.08.07

json是什么
json是什么

JSON是一种轻量级的数据交换格式,具有简洁、易读、跨平台和语言的特点,JSON数据是通过键值对的方式进行组织,其中键是字符串,值可以是字符串、数值、布尔值、数组、对象或者null,在Web开发、数据交换和配置文件等方面得到广泛应用。本专题为大家提供json相关的文章、下载、课程内容,供大家免费下载体验。

547

2023.08.23

jquery怎么操作json
jquery怎么操作json

操作的方法有:1、“$.parseJSON(jsonString)”2、“$.getJSON(url, data, success)”;3、“$.each(obj, callback)”;4、“$.ajax()”。更多jquery怎么操作json的详细内容,可以访问本专题下面的文章。

335

2023.10.13

go语言处理json数据方法
go语言处理json数据方法

本专题整合了go语言中处理json数据方法,阅读专题下面的文章了解更多详细内容。

82

2025.09.10

pip安装使用方法
pip安装使用方法

安装步骤:1、确保Python已经正确安装在您的计算机上;2、下载“get-pip.py”脚本;3、按下Win + R键,然后输入cmd并按下Enter键来打开命令行窗口;4、在命令行窗口中,使用cd命令切换到“get-pip.py”所在的目录;5、执行安装命令;6、验证安装结果即可。大家可以访问本专题下的文章,了解pip安装使用方法的更多内容。

373

2023.10.09

更新pip版本
更新pip版本

更新pip版本方法有使用pip自身更新、使用操作系统自带的包管理工具、使用python包管理工具、手动安装最新版本。想了解更多相关的内容,请阅读专题下面的文章。

436

2024.12.20

C# ASP.NET Core微服务架构与API网关实践
C# ASP.NET Core微服务架构与API网关实践

本专题围绕 C# 在现代后端架构中的微服务实践展开,系统讲解基于 ASP.NET Core 构建可扩展服务体系的核心方法。内容涵盖服务拆分策略、RESTful API 设计、服务间通信、API 网关统一入口管理以及服务治理机制。通过真实项目案例,帮助开发者掌握构建高可用微服务系统的关键技术,提高系统的可扩展性与维护效率。

76

2026.03.11

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 4.9万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.9万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号