windows10下安装pytorch(简单步骤)

最新推荐文章于 2024-06-07 20:29:40 发布

心如花木js

最新推荐文章于 2024-06-07 20:29:40 发布

阅读量280

点赞数 1

分类专栏： pytorch 文章标签： windows pytorch

本文链接：https://blog.csdn.net/qq_45741255/article/details/116423334

版权

1 篇文章 0 订阅

订阅专栏

1.官网安装命令

首先打开pytorch官网，找到你要下载的版本：

主要选择在wind还是Linux、cuda(显卡)还是cpu、下载方式等

这里笔者选择pip安装方式，因为我尝试过conda安装，安装的不成功(有时候是下载慢，跟换镜像后，下载成功但编译器又没法识别)，所以我推荐使用pip方式，下载速度快，而且一次就成功，在编译器中可以直接识别

打开wind下的命令行提示工具(wind + R)，输入cmd

然后将之前选择的下载命令输入命令行中，就开始下载安装了

安装完成之后，在编译器中进行验证，查看是否安装成功

import torch
# import torchvision
print(torch.__version__)#输出PyTorch版本
print(torch.cuda.is_available())

安装成功的话会显示：

经过这么多次安装软件、框架啊什么的，笔者发现还是去官网找寻资料，按照官网的提示是最好的。

(第一次写博客，主要是记录自己的经验和保存记录，以免之后跟换电脑重新安装各种软件遇到问题，提前做好准备，哈哈哈)

关注