Anaconda+vscode+pytorch环境搭建过程详解

1. 安装 Anaconda

Anaconda 是一个用于数据科学的 Python 发行版,集成了常用的科学计算包、数据分析包和机器学习包,方便用户在进行数据处理、建模等工作时直接调用相应的库函数,无需自行搭建环境。以下是安装 Anaconda 的详细步骤:

1.1 下载安装包

访问官网下载页面,选择与您所使用计算机操作系统相符的 Python 版本和 Anaconda 发行版版本,推荐选择最新的版本进行安装。下载完成后,双击下载文件,跟随提示进行安装即可。

1.2 设置环境变量

在安装完成后,需要设置环境变量,以便在命令行中调用 Anaconda 相关命令。具体步骤如下:

1. 打开终端或命令行模式。

2. 输入以下命令:

echo 'export PATH="$HOME/anaconda3/bin:$PATH"' >> ~/.bashrc

3. 输入以下命令,更新环境变量:

source ~/.bashrc

设置完成后,可以输入命令 conda 验证是否安装成功。

2. 安装 Visual Studio Code

Visual Studio Code(简称 VS Code)是一款由 Microsoft 开发的免费、开源的轻量级代码编辑器,适用于 Windows、Linux 和 macOS 等多个平台,拥有丰富的插件和快捷键,是编写代码的好工具。以下是安装 VS Code 的详细步骤:

2.1 下载安装包

访问官网下载页面,选择与您所使用计算机操作系统相符的版本,推荐选择最新的版本进行安装。下载完成后,双击下载文件,跟随提示进行安装即可。

2.2 安装 Python 插件

安装完成后,需要在 VS Code 中安装 Python 插件,以便在编辑 Python 代码时获取语法提示和代码补全功能。具体步骤如下:

1. 打开 VS Code,进入扩展面板,或使用快捷键 Ctrl+Shift+X

2. 在搜索栏中输入 “Python”,选择由 Microsoft 提供的官方 Python 插件,点击 “Install” 安装插件:

Microsoft Python ({*})

3. 安装 PyTorch

PyTorch 是一个基于 Torch 的 Python 开源机器学习库,具有动态神经网络构建功能、支持 GPU 加速、具有一致的 NumPy 表示等优点,广泛应用于计算机视觉、自然语言处理等领域。以下是使用 Anaconda 在 VS Code 中安装 PyTorch 的详细步骤:

3.1 创建虚拟环境

为了避免不同项目之间相互干扰,我们使用 Anaconda 创建一个新的虚拟环境来安装 PyTorch。具体步骤如下:

1. 打开终端或命令行模式。

2. 输入以下命令,创建虚拟环境:

conda create --name myenv

其中,“myenv” 为您所创建的虚拟环境的名称,可自行替换。

3. 激活虚拟环境:

conda activate myenv

4. 在虚拟环境下使用以下命令安装 PyTorch 和 torchvision:

conda install pytorch torchvision torchaudio cpuonly -c pytorch

上述命令中,“cpuonly” 指的是只安装 CPU 版本的 PyTorch,如需安装 GPU 版本,请查看官方文档

3.2 测试 PyTorch 安装

在安装完成后,可以使用以下代码测试 PyTorch 是否成功安装:

import torch

import torchvision

print(torch.__version__)

print(torchvision.__version__)

若输出版本号,则说明 PyTorch 安装成功。

4. 总结

通过本文的介绍,我们详细了解了如何使用 Anaconda 安装 PyTorch,并在 VS Code 中编辑和运行 Python 代码。希望本文可以对您有所帮助。

后端开发标签