Pytorch安装—CPU版(极速版)

2023-12-30 13:40:31

准备工作:

  • 下载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是否安装成功。

文章来源:https://blog.csdn.net/m0_65995252/article/details/135303837
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。