当前位置:   article > 正文

Anaconda 安装 Pytorch_anaconda清华镜像安装pytorch

anaconda清华镜像安装pytorch

下载Anaconda,最新版本的即可,默认安装,最好不要安装在C盘,否则后面C盘容量会很大。

安装Pytorch

打开 Anaconda Prompt ,先切换镜像源为国内清华镜像源,这样安装包的时候下载速度会快一些,也容易成功一些。

在 Anaconda Prompt 命令行依次输入以下四条命令切换到清华镜像源

  1. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  2. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  3. conda config --set show_channel_urls yes
  4. conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

这个镜像源的配置只要一次,之后无需再配置,配置截图如下。

如果想要切换回默认的镜像源,可以使用如下命令。

conda config --remove-key channels

接下来创建一个名为pytorch的环境,并在环境中安装版本为3.9的python,命令如下

conda create --name pytorch python=3.9

完成后再次输入下面的命令来查看已有的 Conda 环境信息。

conda info -e

显示结果如下图,可以看到 pytorch 环境已经添加成功了。

接下来使用下面的命令来激活 pytorch 环境。

conda activate pytorch

接着到 Pytorch官网(点此即可直达Pytorch官网)复制适合自己电脑安装 Pytorch 的命令,如下图所示(我的是之前的版本,需要根据自己的CUDA版本来选择)

所以我的命令是(这里的 -c pytorch去掉,使用镜像安装):

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3

安装过程中输入 y 即可继续安装

等待安装成功即可

本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号