当前位置：首页 > article >正文

windows系统amd核显机器安装cpu版本pytorch（仅用于学习）

article 2025/3/17 20:55:30

文章目录

前置环境
配置镜像源
安装pyrotch

前置环境

安装anaconda环境，并保证环境变量已配置，检测命令

conda --version

配置镜像源

提醒：请务必使用管理员身份运行终端，防止后续操作因权限不够导致的错误。

使用管理员身份运行终端，查看镜像源列表
```
conda config --show channels
```
删除配置的非默认镜像源
```
conda config --remove-key channels
```

设置全局国内镜像源

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

安装pyrotch

提醒：请务必使用管理员身份运行终端，防止后续操作因权限不够导致的错误。

使用anaconda创建虚拟环境
```
conda create -n pytorch_cpu python=3.9
```
执行conda初始化
```
conda init
```
激活虚拟环境
```
conda activate pytorch_cpu
```

安装cpu版的pytorch

pip3 install torch torchvision torchaudio

测试pytorch是否安装成功

import torch
print(torch.__version__)          # 输出版本号
print(torch.cuda.is_available())  # AMD GPU下可能返回False（需ROCm支持）

(pytorch_cpu) PS C:\Users\HP> python
Python 3.9.21 (main, Dec 11 2024, 16:35:24) [MSC v.1929 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
2.6.0+cpu
>>> print(torch.cuda.is_available())
False