使用Anaconda安装pytorch(CPU版)

一、准备

1.Python版本选择

使用Anaconda安装pytorch(CPU版)_第1张图片

在官方文档(https://pytorch.org/get-started/locally/)中显示:目前,Windows 上的 PyTorch 仅支持 Python 3.7-3.9;不支持 Python 2.x。

因此可根据个人需求选择 3.7-3.9 的版本

2.Pytorch选择

使用Anaconda安装pytorch(CPU版)_第2张图片
若电脑为集显,或显卡为AMD,或NVIDIA显卡版本较低直接选择CPU即可,因此我们的环境为:Stable(1.10.2),Canda,Python,CPU

对应的命令为:

conda install pytorch torchvision torchaudio cpuonly -c pytorch

其他环境需求可看:https://pytorch.org/get-started/locally/

二、安装Pytorch

为了方便管理我们的项目,我们首先在Anaconda中创建一个虚环境

1.打开Anaconda Prompt

使用Anaconda安装pytorch(CPU版)_第3张图片

2.创建虚环境

创建python版本为3.8,名为“pytorch”的虚环境

conda create --name pytorch python=3.8

3.激活该环境

activate pytorch

使用Anaconda安装pytorch(CPU版)_第4张图片

4.输入Pytorch安装命令

以下为CPU版的命令,其他版本可参考(https://pytorch.org/get-started/locally/)

conda install pytorch torchvision torchaudio cpuonly -c pytorch

5.验证Pytorch是否安装成功

在命令行中输入:

python

输入:

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

显示以下内容即为安装成功:
使用Anaconda安装pytorch(CPU版)_第5张图片

你可能感兴趣的:(pytorch,python)