Pytorch基础指令

1. pytorch 指定使用GPU卡

##方法一：在运行脚本前，使用 CUDA_VISIBLE_DEVICES 环境变量
export CUDA_VISIBLE_DEVICES=0,1  # 使用 GPU 卡号为 0 和 1
python your_script.py

#或者
CUDA_VISIBLE_DEVICES=0,1 python your_script.py （常用方法）

## 方法二：使用torch.cuda.device
import torch

# 指定使用的 GPU 卡，例如使用 GPU 卡号为 0 和 1
torch.cuda.device(0)  # 使用第一个 GPU
torch.cuda.device(1)  # 使用第二个 GPU

## 方法三：在 PyTorch 中设置 CUDA_VISIBLE_DEVICES
import os
import torch

# 指定使用的 GPU 卡，例如使用 GPU 卡号为 0 和 1 （发现有的时候不生效）
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"

# 在这之后的 PyTorch 操作将在指定的 GPU 卡上执行