安装pytorch

• 配置conda代理

$ cat ~/.condarc
# https://help.mirrors.cernet.edu.cn/anaconda/
ubuntu@ubuntu-Z790-UD:/opt$ cat ~/.condarc
channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.zju.edu.cn/anaconda/pkgs/main
  - https://mirrors.zju.edu.cn/anaconda/pkgs/r
  - https://mirrors.zju.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.zju.edu.cn/anaconda/cloud
  msys2: https://mirrors.zju.edu.cn/anaconda/cloud
  bioconda: https://mirrors.zju.edu.cn/anaconda/cloud
  menpo: https://mirrors.zju.edu.cn/anaconda/cloud
  pytorch: https://mirrors.zju.edu.cn/anaconda/cloud
  pytorch-lts: https://mirrors.zju.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.zju.edu.cn/anaconda/cloud

  nvidia: https://mirrors.cernet.edu.cn/anaconda-extra/cloud/

安装anaconda

sudo bash Anaconda3-2024.02-1-MacOSX-x86_64.sh

• 虚拟环境，base 环境安装pytorch失败

sudo conda create -n pytorch  
conda activate pytorch
sudo conda install -n pytorch  pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 -c pytorch

• spyder

 sudo conda install -n pytorch  spyder-kernels=2.5

• 删除

sudo conda env remove -n pytorch

使用

• python 3.12 有部分包不支持，回退

sudo conda create -n torch python=3.11 pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 -c pytorch
sudo conda install -n torch  spyder-kernels=2.5
sudo conda install -n torch  llvm-openmp

• 新版

sudo conda create -n new python=3.11 pytorch::pytorch torchvision torchaudio -c pytorch

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

维护

sudo conda update -n torch  spyder-kernels
sudo conda update -n base -c defaults spyder conda

资料

• https://pytorch.org/get-started/locally/