Windows10在Anaconda3中安装PyTorch cpu版本

我的电脑已经安装好Anaconda3,如果没安装需要找教程安装下

创建虚拟环境

打开Anaconda Prompt(Anaconda3),以包管理员身份在Windows中设置Python虚拟环境

Windows10在Anaconda3中安装PyTorch cpu版本_第1张图片

# 创建虚拟环境,newName表示虚拟环境名字,python要指定到具体版本
conda create --name newName python=3.7

# 我在命令行中输入如下代码,创建名为pytorch的虚拟环境,python版本为3.9
conda create --name pytorch python=3.9

检查Python和Anaconda版本代码

python --version
conda --version

激活新建的虚拟环境

conda activate pytorch

可以输入conda info --envs查看已有虚拟环境

通过conda安装PyTorch包

进入PyTorch官网:PyTorch,下滑设置自己需要的配置

Package:选择conda

Compute Platform:CPU

Windows10在Anaconda3中安装PyTorch cpu版本_第2张图片

复制自动生成的命令,在Anaconda中运行

conda install pytorch torchvision torchaudio cpuonly -c pytorch

等待安装,测试是否安装成功

输入python,并运行以下代码

import torch
x = torch.rand(2,2)
print(x)

Windows10在Anaconda3中安装PyTorch cpu版本_第3张图片

安装成功!

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