windows系统amd核显机器安装cpu版本pytorch(仅用于学习)
文章目录
- 前置环境
- 配置镜像源
- 安装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