0
点赞
收藏
分享

微信扫一扫

Windows下AMD显卡配置pyTorch记录

穆熙沐 2024-11-11 阅读 24

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)```
举报

相关推荐

0 条评论