创建名为 torch2.7_cu126 的环境

conda create -n torch2.7_cu126 python=3.12 -y

激活环境

conda activate torch2.7_cu126

确保在 torch2.7_cu126 环境下

安装 PyTorch 2.7, torchvision, torchaudio

--index-url 指向 PyTorch 官方的 CUDA 12.6 仓库

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

import torch
import sys

print(f"Python Version: {sys.version}")
print(f"PyTorch Version: {torch.version}")
print(f"CUDA Version (Torch built with): {torch.version.cuda}")
print(f"Is CUDA available? {torch.cuda.is_available()}")

if torch.cuda.is_available():
print(f"Current Device: {torch.cuda.get_device_name(0)}")
# 测试一个小张量计算
x = torch.randn(1).cuda()
print("Tensor on GPU:", x)