Pytorch安装—CPU版(极速版)

本文指导如何在Anaconda中为PyTorch创建虚拟环境,包括使用`condacreate`和`pip3install`下载CPU版本的PyTorch、torchvision和torchaudio,以及提供测试代码示例。
摘要由CSDN通过智能技术生成

准备工作:

  • 下载python
  • 下载安装anaconda

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

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

二、下载pytorch&测试

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

下载前应该指定虚拟环境(我指定的是刚创建的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)

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值