Windows下AMD显卡配置pyTorch记录
Windows下AMD显卡配置pytorch记录
环境：必须使用python3.10.11
配置 Python 环境
1、conda环境配置
conda create -n pytorch python=3.10.11
conda activate pytorch
2、 安装 PyTorch 和 Torch-DirectML
conda install numpy pandas tensorboard matplotlib tqdm pyyaml -y
conda install pytorch torchvision torchaudio cpuonly -c pytorch
pip install opencv-python
pip install wget
pip install torch-directml
3、测试验证
import torch
import torch_directml
# 设置默认设备为 DirectML
device = torch_directml.device()
# 创建一个张量并移动到 DirectML 设备
x = torch.tensor([1.0, 2.0, 3.0], device=device)
# 进行一些操作
y = x + 2
print(y) # 输出: tensor([3., 4., 5.], device=directml:0)```