anaconda下安装pytorch

文章目录

  • 一、安装anaconda
  • 二、安装pytorch
    • 1.打开Anaconda Prompt命令行
    • 2. 使用清华镜像
    • 3.建立环境
    • 4 安装pytorch
  • 5. 验证是否安装成功


一、安装anaconda

官网一般显示的是最新版本:

https://www.anaconda.com/products/individual

如果想要下载以往版本,可以在下面的网页:

https://repo.anaconda.com/archive/

二、安装pytorch

1.打开Anaconda Prompt命令行

在命令行格式下,输入代码,完成调用清华镜像、建立pytorch环境、安装pytorch、测试pytorch过程

2. 使用清华镜像

分别输入以下四个代码

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/

在这里插入图片描述

3.建立环境

建立一个名字为pytorch_z的环境
代码如下(示例):

conda create -n pytorch_z python=3.7

弹出提示之后输入y,即可安装

之后输入以下命令,即可看到已经安装好了pytorch_z

 conda env list

anaconda下安装pytorch_第1张图片
之后切换到自己新建的环境中

conda activate pytorch_z

在这里插入图片描述

4 安装pytorch

根据自己的cuda版本,在pytorch官网寻找代码

输入nvcc -V(注意大写),即可看到cuda的版本

nvcc -V

anaconda下安装pytorch_第2张图片

1)网站: pytorch官网

这里显示只有10.2的,但是我是10.1,所以可以点击
install previous versions of PyTorch
来找到历史版本
anaconda下安装pytorch_第3张图片

这里可以看到我10.1版本的安装命令
anaconda下安装pytorch_第4张图片

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

5. 验证是否安装成功

(1)在命令行左边为 pytorch 环境中,输入 python

(2)之后,输入 import torch,如果没有报错,意味着 PyTorch 已经顺利安装了

(3)接下来,输入 torch.cuda.is_available,如果是 True,意味着你可以使用 GPU,如果是 False,意味着只能使用CPU。
anaconda下安装pytorch_第5张图片

你可能感兴趣的:(人工智能,anaconda,python)