Windows AMD install PyTorch

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

 

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章