根據本機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__