VScode配置PYTORCH环境
时间: 2025-01-08 07:40:00 浏览: 67
### 设置和配置 PyTorch 开发环境
#### 创建并激活虚拟环境
为了确保项目的依赖项隔离,在 Anaconda Prompt 中创建一个新的 Python 虚拟环境。这可以通过执行以下命令来完成:
```bash
Conda create -n pytorch_env python=3.9
```
一旦创建完毕,可以使用如下命令激活该环境[^4]。
```bash
conda activate pytorch_env
```
#### 安装 PyTorch 及其他必要库
在激活后的环境中安装 PyTorch 和任何所需的额外包。具体版本取决于目标硬件支持 CUDA 或仅 CPU 支持。对于大多数情况来说,推荐采用官方提供的预编译二进制文件进行安装:
```bash
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
```
此操作会自动处理所有必要的依赖关系,并将它们添加到当前活跃的 conda 环境中[^1]。
#### VS Code 的集成与配置
打开 Visual Studio Code (VSCode),按 `Ctrl+Shift+P` 组合键唤起命令面板,输入 "Python Select Interpreter" 并选择刚才建立好的名为 `pytorch_env` 的解释器作为工作区默认解释器[^2]。
此时应该能看到状态栏左下角显示所选解释器名称;如果未看到,则需重新启动编辑器使更改生效。
#### 测试安装是否成功
可以在终端里运行简单的 Python 代码片段验证 PyTorch 是否正常运作以及 GPU 加速功能是否可用(如果有)。例如:
```python
import torch
print(torch.cuda.is_available())
```
这段脚本将会打印 True 如果系统能够访问 NVIDIA 显卡用于加速计算;反之则返回 False 表明只启用了 CPU 模式。
阅读全文
相关推荐


















