windows系统amd核显机器安装cpu版本pytorch仅用于学习
目录
windows系统amd核显机器安装cpu版本pytorch(仅用于学习)
前置环境
- 安装anaconda环境,并保证环境变量已配置,检测命令 conda –version
配置镜像源
提醒:请务必使用管理员身份运行终端,防止后续操作因权限不够导致的错误。
- 使用管理员身份运行终端,查看镜像源列表 conda config –show channels
- 删除配置的非默认镜像源 conda config –remove-key channels
- 设置全局国内镜像源 pip config set global.index-url
安装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