Windows AMD install PyTorch

发布时间 2023-11-28 13:59:10作者: alameda

根据本机python版本,创建pytorch环境

conda create -n pytorch python=3.11

激活pytorch环境

activate pytorch

设置清华镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

根据本机环境AMD安装pytorch

conda install pytorch torchvision torchaudio cpuonly


查看envs
conda info –envs

激活pytorch环境
conda activate pytorch

测试pytorch安装是否成功
Python
import torch

 torch.__version__