Pytorch安装—CPU版(极速版)

准备工作:

  • 下载python
  • 下载安装anaconda

一、anaconda创建名为pytorch的虚拟环境

#创建一个名为pytorch的虚拟环境
conda create create -n pytorch
#通过以下语句进行检查:
conda env list

二、下载pytorch&测试

(Pytorch官网现在pip包已经更新为了pip3的下载模式,默认安装最新版本)

Pytorch安装—CPU版(极速版)_第1张图片

下载前应该指定虚拟环境(我指定的是刚创建的pytorch虚拟环境),下载要在此虚拟环境下进行

conda activate pytorch

以下为CPU版本的下载语句,直接复制到命令行里即可:

pip3下载语句

pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple

conda下载语句

conda install pytorch torchvision torchaudio cpuonly -c pytorch -i https://pypi.tuna.tsinghua.edu.cn/simple

下载pytorch后用于测试的代码用例:

>python
>>import torch
>>x = torch.rand(5,3)
>>print(x)

Pytorch安装—CPU版(极速版)_第2张图片

因为下载的是CPU版本的pytorch,所以不能使用torch.cuda.is_available()来检验pytorch是否安装成功。

你可能感兴趣的:(人工智能入门,pytorch,人工智能,python)