Windows下PyTorch环境搭建指南

# Windows 命令行查看显卡信息
nvidia-smi

# 查看已安装的 CUDA 版本
nvcc -V

# 创建新环境
conda create -n pytorch python=3.11
conda activate pytorch
# CUDA 11.8 版本
conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=11.8 -c pytorch -c nvidia
# CUDA 12.1 版本
conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=12.1 -c pytorch -c nvidia
# CUDA 12.6 版本
conda install pytorch torchvision torchaudio pytorch-cuda=12.6 -c pytorch -c nvidia

# CUDA 11.8
pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu118
# CUDA 12.1
pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu121
# CUDA 12.6
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
# CPU 版本（无显卡）
pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cpu

# 清华镜像源
pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 -i https://pypi.tuna.tsinghua.edu.cn/simple
# 安装常用依赖
pip install facenet-pytorch -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install numpy==1.26.4 -i https://pypi.tuna.tsinghua.edu.cn/simple
# 安装 dlib（需要 conda）
conda install -c conda-forge dlib
# 安装 face-recognition
pip install face-recognition -i https://pypi.tuna.tsinghua.edu.cn/simple

# 搜索可用版本
conda search pytorch -c pytorch
conda search pytorch-cuda -c pytorch -c nvidia
# 查看通道配置
conda config --show
# 添加通道
conda config --add channels 通道地址
# 删除环境
conda remove --name 环境名称 --all
conda remove --prefix 路径 --all
# 卸载包
pip uninstall torch torchvision torchaudio
# 查看已安装包
conda list

# 命令行验证
python -c "import torch; print(torch.__version__)"
python -c "import torch; print(torch.cuda.is_available())"

import torch

print(f"PyTorch 版本：{torch.__version__}")
print(f"CUDA 可用：{torch.cuda.is_available()}")

if torch.cuda.is_available():
    print(f"CUDA 版本：{torch.version.cuda}")
    print(f"GPU 数量：{torch.cuda.device_count()}")
    print(f"当前 GPU：{torch.cuda.get_device_name(0)}")

import torch

# 创建随机张量
x = torch.rand(5, 3)
print(x)
print(f"设备：{x.device}")

import torch
import torch.nn as nn

# 检查 CUDA 是否可用
if torch.cuda.is_available():
    device = torch.device("cuda")
    print("✓ 运行在 GPU 上")
else:
    device = torch.device("cpu")
    print("✓ 运行在 CPU 上")

# 创建模型并移动到设备
model = nn.Linear(10, 5).to(device)

# 创建输入数据
x = torch.randn(3, 10).to(device)

# 执行前向传播
output = model(x)

print(f"输出设备：{output.device}")
print(f"输出形状：{output.shape}")
print(f"输出值：\n{output}")

# ❌ 错误写法
device = torch.accelerator.current_accelerator().type if torch.accelerator.is_available() else "cpu"
# ✅ 正确写法 1
device = "cuda" if torch.cuda.is_available() else "cpu"
# ✅ 正确写法 2
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)  # 输出：cuda:0 或 cpu

# 使用清华镜像
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple 包名
# 使用阿里镜像
pip install -i https://mirrors.aliyun.com/pypi/simple/ 包名
# 配置永久镜像源
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

# 先安装 CMake
conda install -c conda-forge cmake
# 再安装 dlib
conda install -c conda-forge dlib
# 最后安装 face-recognition
pip install face-recognition

# 完全卸载
pip uninstall torch torchvision torchaudio
# 清理缓存
pip cache purge
# 重新创建环境
conda remove --name pytorch --all
conda create -n pytorch python=3.11
conda activate pytorch

# 1. 张量操作
import torch
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5, 6])
print(x + y)  # 张量加法
# 2. 自动求导
x = torch.tensor([2.0], requires_grad=True)
y = x ** 2
y.backward()
print(x.grad)  # 输出：4.0
# 3. 神经网络
import torch.nn as nn
class Net(nn.Module):
    def __init__(self):
        super().__init__()
        self.fc1 = nn.Linear(10, 5)
        self.fc2 = nn.Linear(5, 2)
    def forward(self, x):
        x = torch.relu(self.fc1(x))
        x = self.fc2(x)
        return x
model = Net()
print(model)