关于Pytorch#

目前是全球最受欢迎的开源机器学习和深度学习框架之一，最初由前 Facebook的人工智能研究实验室 (FAIR) 开发。

• 特点#

1. Python 风格 -> Pythonic深度整合了 Python 生态，易上手。
2. 应用范围广 -> 目前绝大多数顶级的 AI 论文、前沿研究以及开源模型都优先使用PyTorch实现。
3. 完善的生态 -> 针对不同领域的强大工具包。

Pytorch的环境安装#

安装Conda

• 略

创建虚拟环境

• （Windows下使用Anaconda Prompt 或 Miniconda Prompt）
• 创建一个名为dl的虚拟环境

1
conda create -n dl python=3.10 -y

激活环境

1
conda activate dl

安装 PyTorch

• pip安装：

1
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

• 或者 Conda 官方安装：

1
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

验证安装

• 在终端中输入python进入交互编程后：

1
import torch
2
print(torch.__version__)
3
print(torch.cuda.is_available())
4
print(torch.cuda.get_device_name(0))

python

>>>print(torch.version) 2.6.0+cu124

>>> print(torch.cuda.is_available()) True

>>> print(torch.cuda.get_device_name(0)) NVIDIA GeForce RTX 5080

安装Pycharm并创建 Jupyter项目 需要pycharm专业版

• 如下图新建一个Jupyter 记事本项目，Jupyter的文件后缀为 .ipynb

其他补充命令#

1. conda相关

1
# 退出环境
2
conda deactivate
3
# 删除
4
conda env remove -n dl -y

2. 对于比较新的显卡50系显卡安装Pytorch时采用不同的命令

1
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128

Pytorch 基础语法#