安装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

资料

posted @ 2024-05-03 15:38  bregman  阅读(9)  评论(1)  编辑  收藏  举报