当前位置: 代码迷 >> 综合 >> pytorch 和 torchvision 安装
  详细解决方案

pytorch 和 torchvision 安装

热度:25   发布时间:2024-02-27 03:57:26.0

1. conda 换源(清华源)

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

2. 安装 pytorch 和 torchvision

可以直接去 pytorch 官网找到对应的版本生成对应的命令即可安装。

CPU 版本

conda install pytorch-cpu torchvision-cpu(清华源安装)conda install pytorch-cpu torchvision-cpu -c pytorch (官网安装)