当前位置:   article > 正文

使用Anaconda安装pytorch(CPU版)_用conda安装pytorch为cpu版本

用conda安装pytorch为cpu版本

一、准备

1.Python版本选择

在这里插入图片描述

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

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

2.Pytorch选择

在这里插入图片描述
若电脑为集显,或显卡为AMD,或NVIDIA显卡版本较低直接选择CPU即可,因此我们的环境为:Stable(1.10.2),Canda,Python,CPU

对应的命令为:

conda install pytorch torchvision torchaudio cpuonly -c pytorch
  • 1

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

二、安装Pytorch

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

1.打开Anaconda Prompt

在这里插入图片描述

2.创建虚环境

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

conda create --name pytorch python=3.8
  • 1

3.激活该环境

activate pytorch
  • 1

在这里插入图片描述

4.输入Pytorch安装命令

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

conda install pytorch torchvision torchaudio cpuonly -c pytorch
  • 1

5.验证Pytorch是否安装成功

在命令行中输入:

python
  • 1

输入:

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

显示以下内容即为安装成功:
在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/193838
推荐阅读
相关标签
  

闽ICP备14008679号