目录

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

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

前置环境

  • 安装anaconda环境,并保证环境变量已配置,检测命令 conda –version

配置镜像源

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

  1. 使用管理员身份运行终端,查看镜像源列表 conda config –show channels
  2. 删除配置的非默认镜像源 conda config –remove-key channels
  3. 设置全局国内镜像源 pip config set global.index-url

安装pyrotch

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

  1. 使用anaconda创建虚拟环境 conda create -n pytorch_cpu python=3.9
  2. 执行conda初始化 conda init
  3. 激活虚拟环境 conda activate pytorch_cpu
  4. 安装cpu版的pytorch pip3 install torch torchvision torchaudio
  5. 测试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