以下是搭建 PyTorch 环境的步骤:
- 安装 Anaconda:Anaconda 是一个开源的 Python 发行版本,包含了大量的科学计算库和工具。可以从 Anaconda
官网下载对应操作系统的安装包并安装。 - 创建虚拟环境:为了避免不同项目之间的依赖冲突,可以创建一个虚拟环境。可以使用 Anaconda 中的 conda 命令创建一个名为
myenv 的虚拟环境,命令如下:
conda create -n myenv python=3.9
其中,myenv 为环境名称,python=3.9 表示使用 Python 3.9 版本。创建完毕后,可以使用以下命令激活虚拟环境:
conda activate myenv
- 安装 PyTorch:可以通过 conda 或 pip 安装 PyTorch。如果需要使用 GPU 加速,需要安装对应的 CUDA 和 cuDNN。
• 使用 conda 安装:
conda install pytorch torchvision torchaudio -c pytorch
• 使用 pip 安装:
pip install torch torchvision torchaudio
- 测试 PyTorch 安装:可以通过以下 Python 代码测试 PyTorch 是否安装成功:
import torch x = torch.rand(5, 3) print(x)
运行以上代码,如果能够输出一个随机生成的张量,则表示 PyTorch 安装成功。
- 安装 Jupyter Notebook:Jupyter Notebook 是一个交互式的 Web 应用程序,可以在浏览器中编写和运行
Python 代码。可以通过以下命令安装 Jupyter Notebook:
conda install jupyter
- 启动 Jupyter Notebook:使用以下命令启动 Jupyter Notebook:
jupyter notebook
在浏览器中打开 http://localhost:8888 可以进入 Jupyter Notebook 界面。
这些是搭建 PyTorch 环境的基本步骤。可以根据实际需求安装其他工具和库,如 NumPy、Pandas、Matplotlib 等。